基于Yolov5的绝缘子检测

1.数据集分析

共有两个类别,分别为'insulator','insulator bunch-drop';

两个类别分布如下:

基于Yolov5的绝缘子检测_第1张图片2.如何训练

python train.py  --batch 4 --epochs 100 --data data/insulator.yaml --cfg models/yolov5s.yaml  --weights weights/yolov5s.pt 

各项指标如下:

 基于Yolov5的绝缘子检测_第2张图片

基于Yolov5的绝缘子检测_第3张图片

 基于Yolov5的绝缘子检测_第4张图片

 基于Yolov5的绝缘子检测_第5张图片

你可能感兴趣的:(YOLOV5,深度学习项目落地,python)