yolov5自定义模型训练三

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

yolov5自定义模型训练三_第1张图片

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

yolov5自定义模型训练三_第2张图片

yolov5自定义模型训练三_第3张图片

测试是否可以检测ok

yolov5自定义模型训练三_第4张图片 

yolov5自定义模型训练三_第5张图片 

yolov5自定义模型训练三_第6张图片

yolov5自定义模型训练三_第7张图片 

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

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

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自训练的整体流程

 

 

 

你可能感兴趣的:(AI,YOLO,人工智能,深度学习)