YOLOv7安全帽检测
1、本项目采用
YOLOv7
算法实现对安全帽检测,在几千多张安全帽数据集中训练得到,我们训练了YOLOv7
模型,经评估我们得出了各个模型的评价指标;
2、目标类别数:2
;类别名:person
、hat
;
3、本项目提供了数据集和上述几个训练好的模型的网链接可供下载;
4、安全帽数据集标签格式:VOC
和YOLO
检测结果可视化:
YOLOv7
安全帽检测模型性能指标
Method | P | R | mAP@.5 | mAP@.5:.95: | Time |
---|---|---|---|---|---|
yolov7-person_hat | 0.92 | 0.855 | 0.898 | 0.566 | 4.1 ms |
1、安全帽检测模型训练
数据集文件夹结构目录:
└─person_hat_dataset
├─images
│ ├─test
│ ├─train
│ └─val
└─labe