Mac M1 Yolo5初步尝试:训练自己的模型

287 阅读1分钟

首先得准备一些包含检测目标的图片,比如一个游戏窗口, 得自动持续截图可参考这里: python对屏幕某块区域持续截图

1.在图片里标注目标,可以用labelImg

image.png

image.png

2.标注完后,会在图片同级生成同名的.txt文件,里面放着类似rect的东西

image.png 3.处理图片和标注文件再弄个A.yaml配置来说明这些图片标注,类别

image.png 4.开始训练,参数可以改改

!python train.py --img 640 --batch 50 --epochs 100 --data ../yolo_A/A.yaml --weights yolov5s.pt --nosave --cache

我就弄了37张图片,效果还是有一点的,主要机器太烂,训练花了一个多小时。

val_batch0_labels.jpg 注意,最好验证集的图片不要和训练集的一样