本次是使用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界面,可打开本地文件夹选择要识别的图片
该文介绍了如何运用yolov4目标检测技术进行口罩识别,提供了已标注的口罩数据集、demo仓库以及详细的步骤。通过运行四个Python脚本,包括创建txt文件、训练模型、预测和UI界面交互,实现模型训练和应用。
4075

被折叠的 条评论
为什么被折叠?



