计算机与现代化 ›› 2025, Vol. 0 ›› Issue (09): 35-42.doi: 10.3969/j.issn.1006-2475.2025.09.005
摘要:
摘要:目前基于深度学习的垃圾分类检测算法通常具有大量的模型参数,这会导致存储和计算成本上升,使得在资源受限的移动设备上运行时会产生较高的计算负载。针对以上问题提出一种改进YOLOv8n的轻量化垃圾检测算法。改进算法在YOLOv8n特征提取网络模块中使用GhostNet卷积模块实现轻量化网络;采用RepConv结构重参数化改进主干网络,增强网络特征提取能力的同时减小推理阶段网络的复杂度;改进颈部网络的C2f模块,使用不同尺寸的卷积核获取多尺度特征信息,从而提高模型检测精度;利用迁移学习提高网络泛化能力,并加速模型训练提高模型的检测精度。实验结果表明,改进后的模型参数量和计算量相较于原模型分别降低了26.8%和24.7%,平均检测精度mAP50和mAP50:95分别为98.1%和93.8%。改进算法不仅减少了模型的参数量和计算量,还具有较好的检测精度,能更好地适应移动设备的需求。
中图分类号: