- 博客(18)
- 收藏
- 关注
原创 yolov8训练pose(人体姿态)数据集
在ultralytics/models/yolo/pose路径下新建一个pose_train_my.py,将以下内容复制进去。如果你是训练非人体的目标,需要修改关键点的名字,数量,以及目标的类名;运行pose_train_my.py开始训练,参数可以用我设置的,也可以自己修改。标注完毕后的数据集格式,图片和json放一起。数据集的存放位置如图。*** 可以只标注看得见的关节点***看不见的可以不用标注。运行后得到分割好的数据集,以及用于训练的yaml文件。修改数据集路径(*** 需要用绝对路径 ***)
2024-08-07 18:10:15
1631
4
原创 yolov8训练segment(分割)数据集
在ultralytics/models/yolo/segment,这个路径下新建一个segment_train_my.py,复制以下内容。运行后会生成两个文件夹,seg和txt。其中seg是最终的数据集(已经划分好的)。里面有一个yam文件,用于训练时用。训练好的模型保存在ultralytics/models/yolo/segment/runs,这个路径下。在项目根目录下新建一个segment_json2txt.py,复制以下内容。注意修改路径和参数,修改完运行,开始训练。
2024-08-02 16:11:20
1089
原创 yolov8训练detect(目标检测)数据集
注意,参数设置优先是results = model.train(task='detect',data="datasets/detect/dataset.yaml", model='yolov8n.pt', epochs=300, imgsz=640, batch=16, workers=0, resume=False)这个里面的。根据自己情况修改参数,或直接去ultralytics/cfg/default.yaml路径下修改,是 一样的。至此,数据集转换完成。
2024-08-01 14:51:13
749
原创 yolov8训练旋转目标检测obb数据集
根据自己的情况修改参数,或者前往ultralytics/cfg/default.yaml,这个路径下修改这个yaml文件里面的参数,都是一样的。3.2 在ultralytics/models/yolo/obb下新建一个obb_train_my.py文件。然后运行data_to_yolo.py,运行完后在labels文件夹下会生成train和val两个文件夹。在项目根目录下创建一个roxml_to_dota.py文件,用于转换成dota能用的obb的txt文件。运行前需修改两个位置!**修改图片后缀**
2024-08-01 11:53:45
1710
2
原创 yolov8官方代码训练模型(coco数据集和voc数据集训练)+onnx模型预测
根据官方的解释,pip 的 ultralytics 库包含了 requirements.txt中的所有库。windows11 和 Ubuntu20.04(建议使用 Linux 系统)首先切换到自己建立的虚拟环境安装 pytorch。参照官网,直接使用以下语句即可导入项目所需要的库。train 和 val 的路径就这样写。2.3 yaml文件。
2023-12-04 16:19:30
15379
27
原创 PadleOCR训练自己的ocr模型之数据标注及导出
输入指令之后,在train_data文件夹下会出现以下文件,其中det是用来训练文字检测的数据集,rec是用来训练文字识别的数据集。此时可以把原来的数据集删去或者移到其他地方。在PaddleOCR根目录下建立train_data文件夹,并且将打标签生成的文件和图片放在该文件夹下。打开终端进入PPOCRLabel的文件夹下,输入以下代码进行数据集的划分。此时文字检测和文字识别的数据集就都制作好了。cd 到 PPOCRLabel 目录下。
2023-11-07 17:13:43
3247
2
原创 YOLOv8-pose之tensorrt版本
1、复制TensorRT-8.4.3.1\bin中内容到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin。4、使用pip install xxx.whl安装TensorRT-8.4.3.1文件夹中框出来的。2、复制TensorRT的include文件夹到CUDA的include文件夹。使用如下脚本:(需要安装ultralytics,直接pip安装即可)这个不影响,只是警告,等待几分钟就转换完成了。(一)模型转换遇到的问题。
2023-11-01 15:11:42
1235
2
原创 paddleocr官方模型转onnx推理预测
链接:https://pan.baidu.com/s/1oA8Cr6LcUJ7tqULRR7ss5Q?链接:https://pan.baidu.com/s/1ktrTCM7pgfWQ6ZSM-6mx_g?那么久修改这个地方,预期多少改多少,比如,以上错误久需要改为48。下载后放到项目目录下,pip install 安装即可。转完后放在一个文件夹里,然后下载这个txt文件放在一起。,直接复制全部代码到一个py文件即可。选择对应自己的python版本。其他的遇到什么安装什么。
2023-10-17 17:06:33
739
原创 onnx模型预测环境安装笔记
CPU版本:直接pip安装GPU版本:先查看自己CUDA版本然后在下面的链接去找对应的onnxruntime的版本查询到对应版本,直接pip安装即可,例如。
2023-09-19 14:35:29
900
原创 paddleocr环境安装以及遇到的问题
CPU版本和GPU版本自己选择链接: 官方安装教程官方说的是推荐使用2.0.1+版本以安装最新版(2023.9.18),最新版为pip install paddleocr==2.7.0.2链接:shapely库选择对应自己的python版本复制下面的文件,直接放入虚拟环境中链接:百度网盘提取码: ufr8然后放入自己的虚拟环境的这个位置注意:如果还是报错,那可能是需要安装 Visual Studio。直接安装即可直接安装
2023-09-18 11:42:20
426
原创 PadleOCR训练自己的ocr模型之训练步骤
一、前期准备1、代码2、标注工具二、环境配置库的安装:在文件配置完后,训练后时看缺什么库就安什么库三、文件配置。
2023-06-26 12:47:27
10949
14
原创 Windows下的虚拟机中python程序打包apk
然后把程序复制一下,点击左边栏kivydev目录下找到kivy,在kivy目录下进入accordion。再将"main.py"复制到当前目录下(即**/home/kivydev/kivy/accordion/**目录下)。它会覆盖此目录下的示例——一个main.py文件。因为打包的时候,主程序的名字都是main.py。粘贴我们之前创建的【共享文件夹】位置,将【自动挂载】打勾,点击确定,配置完成。刚安装完的虚拟机界面是这个样子的,是没有任何系统的。找到对应的py文件,放入Windows的共享文件夹就行了。
2022-12-08 11:54:26
3774
6
原创 扩增数据集 图片、xml、json文件
B站的UP主:图像增强软件V2.0,适用于|目标检测|图像分割|图像分类|_哔哩哔哩_bilibili软件下载:Releases · Fafa-DL/Image-Augmentation (github.com)
2022-10-27 16:11:35
805
原创 CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 2.00 GiB total capacity; 960.29 MiB already
input_shape = [320, 320]-----显存不足的话就调小一点(大小为32的倍数)显存不足就调小-----最小为1(有的是最小为2)-------#不能为0#选择是否打开多线程模式 或者 打开几个线程。3、改变num_workers(线程)的数值。2、改变batch_size的大小。1、改变输入图片的大小。错误:训练时显存不足。
2022-09-13 11:15:55
2206
原创 pytorch--GPU环境安装
3、在安装好CUDA和Cudnn之后(详细过程按照置顶的教程走),配置pytorch环境。温馨提示:保证C盘有25G以上的内存,以前准备好系统还原点以防安装错误。0、安装CUDA之前需要安装Visual Studio(2017)1、先看显卡对应的GUDA版本,下载CUDA 版本。2、根据GUDA的版本去找Cudn。4、其他依赖库的安装。视频下有UP主的代码。...
2022-09-01 10:52:33
1096
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人