YOLOv5&YOLOv7打电话检测模型训练
1、本项目使用
YOLOv5-6.0
版本算法训练打电话数据集,文章末尾包含数据集和训练好的打电话检测模型的网盘链接,数据集标签格式为YOLO
和VOC
两种
2、包含YOLOv7
打电话检测训练模型网盘链接
- 在
anaconda
中安装好YOLOv5
的环境 ; - 在官网中下载项目文件,本文采用的代码版本是
YOLOv5-6.0
; - 准备打电话数据集,标签为
YOLO
格式,也就是txt
格式的标签;
数据层面的优化:
为数据集添加特定负例。一般添加相关场景的背景图片,或者添加可能出现的别的类别作为负例。比如实现打电话检测,我们一般认为打电话这类别或者说这个行为的定义是一个人握着手机靠近耳边,或者靠近嘴巴的状态。其他行为比如就单纯拿着手机玩,或者不拿手机的时候等等其他情况都不是打电话的这