目录
一、数据集
有两种,
1、一种是直接的眼部数据集,如
https://blog.csdn.net/Aoulun/article/details/109250430
2、一种是人脸面部数据集,多一步提取人眼特征,再进行识别
(1)基于CNN检测的方法(Fatigue-Driven-Detection-Based-on-CNN)
数据集相当于是视频流抽取的图片数据集,包括眨眼频率计算、PERCLOS 计算、疲劳监测,可以检测图片或者视频
- 配环境
git clone --recursive https://github.com/PengfeiM/Fatigue-Driven-Detection-Based-on-CNN.git
cd /home/yzhang/Fatigue-Driven-Detection-Based-on-CNN
1.python 3.7.1
2.pytorch 1.0.1
(32条消息) Pytorch 各个GPU版本CUDA和cuDNN对应版本_pytorch和cudnn对应版本_风信子的猫Redamancy的博客-CSDN博客
conda install pytorch==1.5.1 torchvision==0.6.1 pytorch-cuda==9.2 -c pytorch
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 成功
3.python-opencv
pip install opencv-python==3.4.2.17
opencv-python版本问题 - 知乎 (zhihu.com)
4.cuda大概可能是8或者9,时间太久记不清了。 不过主要还是显卡-cuda-cudnn-pytorch版本对应吧
- 数据集
将weights和dataset导入文件夹内,weight储存训练权重(ssd300)