轻量化YOLOv5n的高精度垃圾检测算法

来源：计算机工程与应用杂志2023年第10期北京时间：

作者：涂成凤, 易安林, 姚涛, 贺文伟

单位：1.西南交通大学 信息科学与技术学院，成都 611756；2.西南交通大学 烟台新一代信息技术研究院，山东 烟台 264001

摘要：针对现有部署至移动设备或嵌入式设备的生活垃圾检测模型参数量多，计算量大，且识别种类较少等问题，对YOLOv5n目标检测算法进行了轻量化、高精度的优化研究。在YOLOv5n的架构上引入轻量级网络ShuffleNetv2与GhostNet实现了检测网络的轻量化；同时添加注意力机制SE增强特征提取能力，以及引入基于响应的知识蒸馏算法提升定位和分类的准确率，从而提高目标检测精度。实验结果表明，在HGI-30数据集上，优化后的YOLOv5n的参数量和计算量分别减少22.3%和23.3%，检测精度mAP0.5和mAP0.5：0.95分别增加1.6个百分点和2.6个百分点。

关键词：YOLOv5n,轻量级网络,知识蒸馏,生活垃圾分类