本文使用深度学习和Pytorch(PyTorch 2.0.1\Torchvision 0.15.2)实时检测驾驶员的分心行为,并附录完整代码。 检测分心驾驶是现代汽车中最重要的功能之一。无论是自动驾驶汽车还是其它高端汽车,都配备了驾驶员监控系统,以持续跟踪驾驶员的行为。这对确保驾驶员保持目光在道路上,不被其他活动分散注意力至关重要,同时也在很大程度上避免发生事故。 在本文中,我们将利用一个简单的数据集训练图像分类模型,再在模型推理阶段,利用汽车内部仪表板摄像头捕获的视频预测驾驶员正在执行的动作&行为。 State Farm分心驾驶员检测数据集 数据集简介 本文采用的是State