面向轻量化的改进YOLOv7棉杂检测算法

作者：张勇进; 徐健^*; 张明星

来源：计算机应用, 2023, 1-16.

• 棉杂检测
• YOLOv7
• CSP-RepFPN
• 快速
• 轻量化

针对棉纺厂对原棉吞吐量大、检测时间长而常见卷积神经网络无法实现高实时检测的问题，提出基于轻量化改进的YOLOv7模型对原棉杂质的检测算法，旨在快速高效的对棉杂质进行检测。首先通过删减YOLOv7模型冗余的卷积层从而提高检测速度；其次在主干网络内添加Faster-Net卷积降低算法的计算负担，减少特征图的冗余性从而加快检测速度，实现算法的高实时检测；最后通过在颈部网络内运用CSP-RepFPN(Cross Stage Partial Rep Feature Pyramid Networks)重构特征金字塔，增加特征信息流通，减少特征损失，提高检测精度。实验结果表明，改进的YOLOv7模型在棉杂检测精度上达到了96.0%，检测时间降低了37.5%，在公开DWC(Drinking Waste Classification)数据集上整体精度达到82.5%，检测时间仅为29.8ms，改进的YOLOv7模型能够为原棉杂质的实时检测和识别分类提供一种轻量化的检测方法，大幅节约了时间成本。

• 单位
  西安工程大学