基于轻量级YOLOv8-ASP的航空行李检测

Aviation baggage detection based on lightweight YOLOv8-ASP

  • 摘要: 航空行李具有种类数量繁多、纹理图案复杂和形状各异等特点,这些特点使得深度学习模型需要更高的计算复杂度才能对行李目标进行准确识别。为了增强工程实用性和降低模型复杂度,该文提出了一种基于轻量级YOLOv8-ASP的航空行李检测方法。首先,为优化模型的特征提取和融合能力,同时降低模型复杂性和计算量,设计了自适应Single PAN特征融合网络模块;其次,为增强对航空行李重要特征的感知能力,在YOLOv8n主干网络的底部加入ECA注意力机制;最后,为解决航空行李目标尺度变化引起的不平衡问题,在检测头中引入MPDIoU边框损失函数。通过构建航空行李数据集进行实验验证,结果表明,相对于YOLOv8n原始模型,所提YOLOv8-ASP方法在mAP@50上提高了0.4%,FPS提高了16%,参数量从3.012×106减少到2.759×106,GFLOPs从8.2减少到5.6。实验结果表明,所提方法在保证检测精度的同时,显著降低了模型的复杂性和计算量。

     

    Abstract: Aviation luggage detection presents several challenges due to the wide variety of types, complex textures, and varying shapes of luggage, these challenges require deep learning models to have higher computational complexity to accurately identify the luggage targets. To improve the practical applicability and reduce model complexity, this paper proposes an aviation luggage detection method based on the lightweight YOLOv8-ASP. First, to optimize feature extraction and fusion while reducing model complexity and computation, an adaptive single PAN (path aggregation network) feature fusion network module is designed. Secondly, to enhance the perception of important features for aviation luggage, an efficient channel attention (ECA) mechanism is incorporated to the bottom of the YOLOv8n backbone network. Finally, to address the imbalance caused by scale variations of luggage targets, the MPDIoU bounding box loss function is introduced into the detection head. Experimental validation on an aviation baggage dataset demonstrates that, compared to the original YOLOv8n model, the proposed YOLOv8-ASP method achieves a 0.4% improvement in mAP@50, a 16% improvement in FPS, while reducing the parameter count from 3.012×106 to 2.759×106 and GFLOPs from 8.2 to 5.6. The experimental results show that the proposed method significantly reduces model complexity and computational cost while maintaining detection accuracy.

     

/

返回文章
返回