008yolov4口罩目标检测识别

这篇博客介绍了使用YOLOv4进行口罩目标检测的详细过程,包括数据集、训练模型和预测。提供了包含训练、预测代码的完整Demo,并分享了B站上的演示视频链接。读者可以通过运行指定的Python脚本来实现口罩的自动识别。
摘要由CSDN通过智能技术生成

目标检测一般是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界面,可打开本地文件夹选择要识别的图片

​欢迎下载

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值