基于深度学习实现驾驶员分心驾驶行为识别项目源码+数据集+模型+项目说明.7z
附数据集下载链接,输入一张图片,输出驾驶员状态及概率
驾驶员状态识别如下
c0: 安全驾驶
c1: 右手打字
c2: 右手打电话
c3: 左手打字
c4: 左手打电话
c5: 调收音机
c6: 喝饮料
c7: 拿后面的东西
c8: 整理头发和化妆
c9: 和其他乘客说话
【使用工具】
OpenCV
Matlibplot
Pytorch
TensorboardX
【代码介绍】
data_mean.py 统计训练图片的均值与标准差
splite_valid.py 分离验证集与训练集
visual_classes.py 浏览每个驾驶状态
visual_samples.py 浏览随机的样本
model_plot.py 利用_tensorboardX_进行模型的绘制........