上传者: 42173218
|
上传时间: 2022-02-18 18:15:34
|
文件大小: 189.55MB
|
文件类型: -
运行环境:
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 损失函