基于改进YOLOv4-Tiny算法的移动端实时司机违章行为检测

摘要

司机违章行为在日常生活中屡见不鲜,为解决在移动端上高精度、实时监测司机的驾驶行为的问题,基于一种轻量级的目标检测算法YOLOv4-tiny,通过引入跨通道部分的连接机制,减少了模型的参数,在YOLOv3-tiny的基础上改进残差的思想,并在损失函数上将定位损失替换为CIOU损失实现更精确的边框,最后通过知识蒸馏的手段,利用教师模型YOLOv4指导tiny训练进一步提升其性能。通过在公开数据集和自建数据集上的多项实验对比结果显示,YOLOv4-tiny达到了更高的精度(分别提升7%和10%),实现了对算力较低的嵌入式设备或移动端的实时检测。