1.修改官方的权重文件,包含yolov5s.pt、yolov5m.pt、yolov5l.pt、yolov5x.pt
我们用的是s.pt,打开文件将nc改为自己训练的种类数量。
打开train.py,路径YOLOV5\train.py,可以按照下面的提示修改适合自己的参数
然后先运行一下,如果有报错的话,按报错说的改。有缺少的库就安装,注意有些库版本是否能用,一定要仔细看。如果没有报错的话会在YOLOV5\runs\train\exp6\weights下出现两个.pt文件。自己找找,可能路径会不一样。
然后就可以检测自己的模型了,找到detect.py,路径在:\YOLOV5\detect.py
然后将官方的模型改成自己的模型,就可以显示了。这里我调用的是摄像头,所以将source的默认值改为0,如果要调用视频的话,可以改为视频的路径。
最后跑出来的效果是这样的:
虽然权重不高,但是40张照片训练出来的 ,可以跑通我觉得也很OK了