目标检测一般是yolov3、yolov4、yolov5、yolox、PSPnet、faster_rcnn、SDD等
本次是使用yolov4目标检测做口罩的识别,包括已经做好标签的口罩数据集和整个代码的运行可以看bilibili
https://space.bilibili.com/124080712?spmidfrom=333.999.0.0
上效果图:
完整的demo文件
数据集放在了dataset文件夹下,有口罩原图和对应的xml标签文件
依次运行demo 01、02、03、04即可
01create_txt.py是对标签文件生成train.txt保存路径在黎明
02train.py是将txt中的路径读取训练模型保存在weights文件夹下
03predict.py是调用模型对单张图片的预测
04pyqt界面.py是可交互的ui界面,可打开本地文件夹选择要识别的图片
欢迎下载