驾驶员手机使用检测模型:优化Yolov5n算法

摘要：

为进一步实现在移动设备或嵌入式设备上对手机使用的违法行为进行实时检测,通过优化Yolov5n算法提出了一种轻量化、高精度、实时性的检测模型.将Focal-EIoU Loss与FocalL1 Loss相结合来获得更加精确的框定位以及损失函数的更快收敛.利用Slimming剪枝算法来进一步提高模型的轻量化及计算效率.在模型微调时利用数据增强技术对微调操作进行指导,从而使模型能够获得更好的性能提升.在手机使用数据集上对改进方法进行消融实验,进一步验证检测模型的有效性.实验表明,优化后的模型在手机使用数据集及Pascal VOC 2012数据集上的检测精度分别提高了0.2、12.3个百分点,参数量减少44.4%,计算量分别减小45.2%、40%,有利于模型进一步在移动设备及嵌入式设备上的实时性检测.

关键词：Yolov5n算法优化、Slimming剪枝、Focal-EIoU Loss、FocalL1 Loss、数据增强

所属期刊栏目：59

分类号：TP391(计算技术、计算机技术)

资助基金：国家自然科学基金;内蒙古自治区科技成果转化专项资金项目;内蒙古自治区科技成果转化专项资金项目;内蒙古自然科学基金项目;内蒙古自治区研究生科研创新项目

在线出版日期：2023-09-28（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：129-136