1、前言
2、运行YOLOv10代码流程(超详细)
(3)根据下面步骤安装:
(4)数据集和其他配置
(5)测试训练(很详细)
1、前言
由于一些事情,之前跑的YOLOv8的一个实验,投了EI会议。
最近基本上都是,等待审核,和返修。准备开始写第二篇论文,那肯定也必须是YOLOv10了。今天下午跑了下代码,很顺畅。写个博客记录下流程。
2、运行YOLOv10代码流程(超详细)
(1)YOLOv10的github地址:https://github.com/THU-MIG/yolov10?tab=readme-ov-file
(2)将代码先下载下来:
方式1:直接下载压缩包
方式2: git clone https://github.com/THU-MIG/yolov10.git
(3)根据下面步骤安装:
① 新建一个conda环境:conda create -n yolov10 python=3.9
② 激活新建的环境: conda activate yolov10 (这里说明下,我在Linux上面只用所说的命令激活不成功,用的是 source activate yolov10 才激活成功的。根据自己的情况选择就行)
③ 安装所需要的:pip install -r requirements.txt
④ 运行:pip install -e
到这里,环境基本上就搭建好了。
(4)数据集和其他配置
我们一般都是使用的coco128数据集,直接可以在网上下载。
然后到coco128.yaml里面修改下地址就行:地址改为你将coco128数据集下载的位置。
coco128.yaml文件位置:xxx/yolov10/ultralytics/cfg/datasets/coco128.yaml
(5)测试训练(很详细)
yolo detect train data=coco128.yaml的位置地址 model=yolov10n/s/m/b/l/x.yaml文件的位置 epochs=500 batch=32 imgsz=640 device=0
以我的为例子:
yolo detect train data=coco128.yaml model=yolov10n.yaml epochs=300 batch=32 imgsz=640 device=0
(如果你没有改建.yaml文件夹的位置,可以直接写文件名称,YOLO系列项目都有相对位置设置的)
接下来,有的人可能会报错,如下所示(对,说的就是我,我运行的时候报错了):
这个不是什么大问题,小小的问题,改个相对路径就行。
找到:xxx/yolov10/yolov10/ultralytics/models/yolov10/train.py
保存后,重新运行,就可以正常啦。
至此,YOLOv10就成功运行起来啦。撒花✿✿ヽ(°▽°)ノ✿
剩下的,就靠自己缝补咯。