ssd处理灰度图_My-Fatigue-Driven-Detection-Based-on-SSD: （1）训练出监测驾驶员精神状态的模型，能够运用在车载摄像头中，对疲劳的驾驶员发出警报。 （2）搭建...

运行环境：

1.python 3.7.1

2.pytorch 1.0.1

3.python-opencv

说明

预训练的权重文件[vgg_16]

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

训练运行python Train.py

单张测试 python test.py

测试网络性能 python eval.py

测试视频 python camera_detection.py

##目前进度：

1、PERCLOS计算DONE

2、眨眼频率计算DONE

3、打哈欠检测及计算DONE

4、疲劳检测DONE

主要文件说明：

ssd_net_vgg.py定义class SSD的文件

Train.py 训练代码

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

loss_function.py损失函数

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

eval.py评估网络性能代码

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

l2norm.pyl2正则化

Config.py配置参数

utils.py工具类

camera.pyopencv调用摄像头测试

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

video_detection.py视频检测，V3

数据集结构：

/dataset:

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

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

/JPEGImages存放图片

/gray2rgb.m灰度图转三通道

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

权重文件存放路径：

weights

测试后的图片存放位置：

tested

1、运行Train.py训练

2、eval可以用于测试整个测试集，test用于单张图片测试。