ultralytics-yolov8-训练自己的数据集(以neu-det为例)
·
一、准备yolo格式的数据集

这个是我在github找的neu-det的yolo格式的数据集,是别人已经转换好的,原先网上能下载的都是coco格式的数据集,自己转换比较麻烦
链接:GitHub - KingRedMan/NEU-DET_Yolo: 铁轨缺陷检测数据集NEU-DET的Yolo格式
(直接搜neu-det yolo格式都可以搜的出来的)
二、修改net-det.yaml
NEU-DET_Yolo-master/NEU-DET_Yolo-master/neudet.yaml这个是我自己的路径,找到自己的数据集下的yaml文件,一般都是展开就能看到的,且只有一个,很好找

点开之后,把这三个路径换成自己的
三、开始训练
也可以是终端也可以是新建一个train.py文件
终端的话就是指令:yolo train model=yolov8.yaml data=NEU-DET_Yolo-master/NEU-DET_Yolo-master/neudet.yaml epochs=200
新建一个train.py文件的话,就是这样的,直接gpt或者gemi生成一个就好

更多推荐
所有评论(0)