首先得准备一些包含检测目标的图片,比如一个游戏窗口, 得自动持续截图可参考这里: python对屏幕某块区域持续截图
1.在图片里标注目标,可以用labelImg
2.标注完后,会在图片同级生成同名的.txt文件,里面放着类似rect的东西
3.处理图片和标注文件再弄个A.yaml配置来说明这些图片标注,类别
4.开始训练,参数可以改改
!python train.py --img 640 --batch 50 --epochs 100 --data ../yolo_A/A.yaml --weights yolov5s.pt --nosave --cache
我就弄了37张图片,效果还是有一点的,主要机器太烂,训练花了一个多小时。
注意,最好验证集的图片不要和训练集的一样