参考:https://zhuanlan.zhihu.com/p/604829541
1.数据集
自己要有数据集:
2.进入roboflow官网,链接是检测模型用的:https://app.roboflow.com/tom-skp3p/createSample,命名后将上面dataset直接上传folder,点击yolov8选项后,选择train\test\val比例,自动导出(这一步实现了xml到txt转换),同时生成完整数据集如下,直接放在第一级子目录下:
3.修改data.yaml文件:将第一行:直接绝对路径
4.修改后跑代码
yolo task=detect mode=train model=yolov8n.pt data=tomatodata.yaml epochs=100 imgsz=640 workers=4 batch=4