运行环境:
1.python 3.7.4 2.pytorch 1.4.0
3.python-opencv
说明
预训练的权重文件[vgg_16] 具体的配置文件请看Config.py文件
训练运行python Train.py
单张测试 python Test.py
##目前进度: 1、PERCLOS计算 DONE 2、眨眼频率计算 DONE 3、打哈欠检测及计算 DONE 4、疲劳检测 DONE 5、人脸情绪检测 DONE 6、口罩检测Done
网络检测性能:准确率82.18%
主要文件说明: ssd_net_vgg.py 定义class SSD的文件 Train.py 训练代码 voc0712.py 数据集处理代码(没有改文件名,改的话还要改其他代码) loss_function.py 损失函数 detection.py 检测结果的处理代码,将SSD返回结果处理为opencv可以处理的
2021-09-13 19:40:36
212.52MB
Python
1