基于Pytorch的驾驶员分心行为实时检测

本文使用深度学习和Pytorch(PyTorch 2.0.1\Torchvision 0.15.2)实时检测驾驶员的分心行为,并附录完整代码。

基于Pytorch的驾驶员分心行为实时检测_第1张图片
基于Pytorch的驾驶员分心行为实时检测_第2张图片
基于Pytorch的驾驶员分心行为实时检测_第3张图片

检测分心驾驶是现代汽车中最重要的功能之一。无论是自动驾驶汽车还是其它高端汽车,都配备了驾驶员监控系统,以持续跟踪驾驶员的行为。这对确保驾驶员保持目光在道路上,不被其他活动分散注意力至关重要,同时也在很大程度上避免发生事故。

基于Pytorch的驾驶员分心行为实时检测_第4张图片

在本文中,我们将利用一个简单的数据集训练图像分类模型,再在模型推理阶段,利用汽车内部仪表板摄像头捕获的视频预测驾驶员正在执行的动作&行为。

State Farm分心驾驶员检测数据集

数据集简介

本文采用的是State

你可能感兴趣的:(深度学习开发实战,pytorch,人工智能,python)