本研究运用PyTorch框架及ResNet18架构,结合大规模数据集(含22,400幅多类图片),开发了高效的驾驶员注意力分散检测系统。
分心驾驶员检测项目使用了RESNET18模型和PYTORCH框架进行机器学习研究,旨在解决由于智能设备使用的增加而导致的交通事故问题。
我们基于StateFarm在Kaggle挑战赛中提供的数据集进行了实验:
- 训练数据包含22,424张图像,并被分为十个带有标签的文件夹。
- 测试数据包括7.97万张未标记的图像,用于通过kaggle平台进行准确性的评估。
每一张提供给我们的图像是640×480像素大小。为了适应ResNet18模型的需求,我们对输入图片进行了预处理,并在PyTorch框架中对该网络做了一些调整。
实验过程中使用了交叉熵损失函数和SGD优化器,设置了学习率为0.001、动量为0.9以及Nesterov动量。训练阶段我们的精度达到了约98%,而测试准确率则接近97%。
我们详细记录并汇报了整个项目流程及实验结果,并制作了相应的幻灯片用于展示研究成果。