Yolo-tiny之口罩快速检测识别

Yolov3_tiny之口罩快速检测

因为需要配置一个轻量化框架,原来使用的yolov3跑起来在10帧左右,运行不是很流畅,今天又重新配置了一下yolov3-tiny版,运行检测帧在20左右,效果不错,也把近期比较热的口罩检测实战一了下,效果还可以。
先看结果和帧数:
在这里插入图片描述

数据集获取

数据集大约在500个,通过下载的小视频,然后写了几行代码截帧完成的,完成后放进JPEGImages中,通过标注软件进行标注完成,框架如下:
在这里插入图片描述

配置修改

修改cfg中yolo的class为2,filters为21
在这里插入图片描述
修改coco_classes.txt 和voc_classes.txt,内容为masking和no_masking
在这里插入图片描述
运行test.py在main下生成相关txt文件
在这里插入图片描述
修改voc_annotation.py中classes为masking和no_masking
在这里插入图片描述
运行voc_annotation.py生成相应txt文件
在这里插入图片描述

训练模型

修改train中相关参数,运行train.py
在这里插入图片描述

如若进行微调继续训练,修改load_pretrained为True
在这里插入图片描述

检测结果

修改yolo.py中相关路径及相关参数,运行yolo_ideo.py进行检测,效果如下:
在这里插入图片描述
在这里插入图片描述

发布了51 篇原创文章 · 获赞 51 · 访问量 9728

猜你喜欢

转载自blog.csdn.net/weixin_44747240/article/details/104532035