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

Yolov3_tiny之口罩快速检测

  • 数据集获取
  • 配置修改
  • 训练模型
  • 检测结果

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

数据集获取

数据集大约在500个,通过下载的小视频,然后写了几行代码截帧完成的,完成后放进JPEGImages中,通过标注软件进行标注完成,框架如下:
Yolo-tiny之口罩快速检测识别_第2张图片

配置修改

修改cfg中yolo的class为2,filters为21
Yolo-tiny之口罩快速检测识别_第3张图片
修改coco_classes.txt 和voc_classes.txt,内容为masking和no_masking
Yolo-tiny之口罩快速检测识别_第4张图片
运行test.py在main下生成相关txt文件
Yolo-tiny之口罩快速检测识别_第5张图片
修改voc_annotation.py中classes为masking和no_masking
Yolo-tiny之口罩快速检测识别_第6张图片
运行voc_annotation.py生成相应txt文件
在这里插入图片描述

训练模型

修改train中相关参数,运行train.py
Yolo-tiny之口罩快速检测识别_第7张图片

如若进行微调继续训练,修改load_pretrained为True
Yolo-tiny之口罩快速检测识别_第8张图片

检测结果

修改yolo.py中相关路径及相关参数,运行yolo_ideo.py进行检测,效果如下:
Yolo-tiny之口罩快速检测识别_第9张图片
Yolo-tiny之口罩快速检测识别_第10张图片

你可能感兴趣的:(目标识别,深度学习,python,深度学习,人工智能,图像识别,yolo)