一、准备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生成一个就好

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐