Skip to content

NiuCoder/distracted_driver_detection

Repository files navigation

distracted_driver_detection(走神司机检测)

本项目是2016年Kaggle上的一个竞赛,目的在于对连续采集的将近80000张图像进行分类,项目的评判标准是所有图像的logloss。本文提供了一种解决方案,利用vgg16、vgg19、resnet50作为预训练模型,然后进行模型集成,最终在Leader board上的排名为110/1440。

运行环境

本项目运行的环境为Intel Core i7-8700K + Nvidia Geforce GTX 1080Ti + 32G内存 + 1T硬盘,项目依赖库详见environment.yaml。

其他说明

  1. 代码及运行结果见distracted_driver_detection.ipynb。
  2. 利用单个预训练模型训练10个分类器的时间约6小时,单个模型的预测时间为1.5小时,三种预训练模型的耗时接近。
  3. 数据集的获取可以从Kaggle官网比赛页获取,注意科学上网,实测在aws上秒下。也可以访问我的网盘, 密码:javp。

About

A kaggle competition for spot distracted drivers

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors