python检测吸烟的算法_GitHub - DohaerisT/DangerousDrivingDetector: 利用SSD目标检测算法判断是否闭眼或者张开嘴和吸烟打电话等手势行为，通过PERCL...

运行环境：

1.python 3.7.4

2.pytorch 1.4.0

3.python-opencv

说明

预训练的权重文件[vgg_16]

具体的配置文件请看Config.py文件

训练运行python Train.py

单张测试 python Test.py

测试视频 python camera_detection.py

##目前进度：

1、PERCLOS计算DONE

2、眨眼频率计算DONE

3、打哈欠检测及计算DONE

4、疲劳检测DONE

5、人脸情绪检测 DONE

网络检测性能：准确率82.18%

主要文件说明：

ssd_net_vgg.py定义class SSD的文件

Train.py 训练代码

voc0712.py数据集处理代码(没有改文件名，改的话还要改其他代码)

loss_function.py损失函数

detection.py检测结果的处理代码，将SSD返回结果处理为opencv可以处理的形式

test.py单张图片测试代码 Ps:没写参数接口，所以要改测试的图片就要手动改代码内部文件名了

l2norm.pyl2正则化

Config.py配置参数

utils.py工具类

camera_detection.py摄像头检测代码V1,V2

augmentations.py 生成识别框

Run.py 主程序运行文件

MainWindow.py UI界面布局

/Emoji 人脸情绪识别部分

/mtcnn 人脸定位检测

/test 测试视频、图片素材

数据集结构(没有给出，需要的话咨询btzheng1018@qq.com)：

/dataset:

/Annotations存放含有目标信息的xml文件

/ImageSets/Main存放图片名的文件

/JPEGImages存放图片

/txt.py生成ImageSets文件的代码

权重文件存放路径：

/weights