因为需要配置一个轻量化框架，原来使用的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进行检测，效果如下：