YOLOv5制作自己的数据集修改参数

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了