
PyTorch_DistractedDriverDetection: Kaggle上的【State Farm】分心驾驶检测竞赛...
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本项目参与了Kaggle平台举办的State Farm分心驾驶检测竞赛,利用PyTorch框架开发模型,旨在识别驾驶员是否处于分心状态。
### 项目概述
#### 1.1 项目来源:
该项目源自Kaggle平台上的【State Farm Distracted Driver Detection】比赛。
#### 1.2 问题描述:
本任务要求对驾驶员行为的图片进行分类,共有10个类别:安全驾驶、左手打字、右手打电话等。
#### 1.3 解决方案思路:
采用预训练好的ResNet-34模型并对其进行微调(finetune)以适应当前数据集和任务需求。
### 程序运行相关
#### 2.1 运行环境
Windows操作系统,Python版本为3.5,PyTorch版本0.4以及Visdom工具。
#### 2.2 准备工作:
s1: 将本项目代码库下载到本地。
s2: 在当前目录下建立如下文件夹,并将训练数据集下载至data文件夹内:
- data
- train
- trained_models
#### 2.3 运行步骤:
s1:启动Visdom后台服务,命令为`pyt`。
全部评论 (0)
还没有任何评论哟~


