EAAI Journal 2026 Journal Article
BeltTear-seg: A lightweight model for belt tear segmentation with multi-scale feature squeeze attention and enhanced classification decision
- Li Yuan
- Hebin Zhou
- Yuqi Kong
- Li Liu
- Jiangyun Li
To address the challenges of small target tear omission and false detections, complex background interference, and strict real-time requirements in belt tear segmentation for industrial production, this study proposes an improved instance segmentation model based on You Only Look Once version 8 nano (YOLOv8n) for segmentation (YOLOv8n-seg), named BeltTear-seg. First, a Multi-scale Feature Squeeze Attention mechanism (MFSA) is introduced to enhance the model’s capability in capturing small target tear regions, effectively reducing the omission rate. Second, an Enhanced Classification Decision (ECD) layer is incorporated into the model head, working in conjunction with Bidirectional Feature Pyramid Network(BiFPN) to reduce the false detection rate. Finally, a lightweight designed Cross Stage Partial with 2 convolutions and feature fusion (Light-C2f) is introduced to significantly enhance computational efficiency while maintaining high segmentation accuracy. Experimental results demonstrate that the BeltTear-seg model performs exceptionally well on a custom-built belt tear instance segmentation dataset, which includes 4, 050 images (1, 800 negative and 2, 250 positive samples) of conveyor belt surfaces collected from real industrial sites, achieving a classification accuracy of 95. 9%, which represents a 5. 8% improvement over the original model YOLOv8n-seg on the custom dataset. Meanwhile the model’s detection time per frame is shortened by 15. 1%. When compared with other mainstream instance segmentation models, this model also demonstrates significant advantages. The proposed improvements not only significantly enhance the accuracy of the instance segmentation model but also simultaneously boost segmentation speed, thereby meeting the dual industrial requirements of high precision and real-time detection for belt tear inspection.