本文提出了一种基于深度学习的安全带检测方法,尝试将近年来在图像检测方面有较好应用的深度学习方法来提高安全带检测的准确率。相比传统的安全带检测方法,深度学习的最大优势在于它可以自动的从样本数据中学习特征,最大限度的减少了人为的干预以及手工设计特征的复杂性。
本方法充分发挥深度学习端到端的特性,充分利用rcnn和yolo系列算法的优点,搭建深度学习中的卷积神经网络模型,并用卷积神经网络模型对样本图片进行训练得到用于安全带检测的模型,然后用该模型对待检测图片进行检测,判断出车辆司机是否佩带安全带。本文中利用Keras框架对整个训练和检测过程进行了实现,并对实验结果进行了分析,证明了该方法的有效性。
1