PyTorch_DistractedDriverDetection
1 项目概述:
1.1 项目来源:
kaggle中的【State Farm Distracted Driver Detection】比赛的实现。
1.2 问题概述:
对一张关于驾驶员行为的图片进行分类,共10类:安全驾驶/左手打字/右手打电话/左右打电话/...
1.3 问题解决思路:
采用resnet34进行finetune
1.4 数据集下载地址:
2 程序运行相关:
2.1 运行环境:
windows/python3.5/pytorch0.4/visdom
2.2 运行前的准备:
s1:下载本repository至本地;
s2:在本目录下,建立文件夹如下,并下载train数据集至data文件夹下:
----data
--train
----trained_models
2.3运行:
s1.打开visdom后台:pyt
2021-09-24 10:13:34
182KB
Python
1