本次是使用yolov4目标检测做口罩的识别,包括已经做好标签的口罩数据集和demo仓库和运行可以看bilibili找008期
到此一游7758258的个人空间_哔哩哔哩_bilibili
上效果图:


完整的demo文件

数据集放在了dataset文件夹下,有口罩原图和对应的xml标签文件
依次运行demo 01、02、03、04即可
01create_txt.py是对标签文件生成train.txt保存路径在黎明
02train.py是将txt中的路径读取训练模型保存在weights文件夹下
03predict.py是调用模型对单张图片的预测
04pyqt界面.py是可交互的ui界面,可打开本地文件夹选择要识别的图片