yolov5自定义模型训练三

经过11个小时cpu训练完如下

在runs/train/expx里存放训练的结果，

测试是否可以检测ok

 

 

 

网上找的这张识别效果不是很好，通过加大训练次数和数据集的话精度可以提升。

训练后的权重也可以用视频源来识别，

python detect.py --source 0  # webcam
                          file.jpg  # image 
                          file.mp4  # video
                          path/  # directory
                          path/*.jpg  # glob
                          'https://youtu.be/NUsoVlDFqZg'  # YouTube
                          'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream

至此实际完整走过一次yolov自训练的整体流程