kaggle中的【State Farm Distracted Driver Detection】比赛的实现。https://www.kaggle.com/c/state-farm-distracted-driver-detection
对一张关于驾驶员行为的图片进行分类,共10类:安全驾驶/左手打字/右手打电话/左右打电话/...
采用resnet34进行finetune
https://pan.baidu.com/s/1l2RkXYOAdxuL_jiKWJm1cQ
windows/python3.5/pytorch0.4/visdom
s1:下载本repository至本地;
s2:在本目录下,建立文件夹如下,并下载train数据集至data文件夹下:
----data
--train
----trained_models
s1.打开visdom后台:python -m visdom.server
s2.运行训练文件:python train.py