
利用PyTorch和ResNet18模型,并借助22,400张不同类型的注意力分散驾驶员图像,构建了一种用于检测的系统。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
CS175分心驾驶员检测,基于ResNet18模型 #PYTORCH。机器学习在分心驾驶领域的应用日益受到重视,尤其是在导致致命事故数量从2014年到2018年显著增加的因素中扮演着关键角色。我们关注的是智能设备普及带来的影响。针对建筑物探测器的数据,StateFarm在他们的Kaggle挑战赛中提供了宝贵资源:训练数据集包含22,424张带有明确标签的图像,这些图像被组织成10个不同的课程文件夹。同时,测试数据集则包含7.97万张未标记的图像,通过Kaggle平台进行准确性评估。每张图像的尺寸均为640×480像素。我们的模型采用ResNet18架构,并对其进行了必要的预处理以及对PyTorch框架的定制化修改。为了优化模型性能,我们采用了交叉熵损失函数和SGD优化器,学习率设定为0.001,动量参数设置为0.9,并结合Nesterov动量技术。训练过程中的精度达到了约98%,测试准确率也接近97%。以下是对我们所采取的操作和报告内容的详细阐述:幻灯片详情可参考://docs
全部评论 (0)
还没有任何评论哟~


