分割
计算机科学
结构工程
材料科学
人工智能
工程类
作者
Wenxuan Zhu,Bochao Su,Xinhe Zhang,Long-yuan Li,S. S. Fang
出处
期刊:Buildings
[MDPI AG]
日期:2024-07-03
卷期号:14 (7): 2036-2036
标识
DOI:10.3390/buildings14072036
摘要
Aluminum profiles are widely used in various manufacturing sectors due to their flexibility and chemical properties. However, these profiles are susceptible to defects during manufacturing and transportation. Detecting these defects is crucial, but existing object detection models like Mask R-CNN and YOLOv8-seg are not optimized for this task. These models are large and computationally intensive, making them unsuitable for edge devices used in industrial inspections. To address this issue, this study proposes a novel lightweight instance segmentation model called AL-damage-seg, inspired by the YOLOv8n-seg architecture. This model utilizes MobileNetV3 as the backbone. In YOLOv8n-seg, the role of C2f is to enhance the nonlinear representation of the model to capture complex image features more efficiently. We upgraded and improved it to form multilayer feature extraction module (MFEM) and integrates a large separable kernel attention (LSKA) mechanism in the C2f module, resulting in C2f_LSKA, to further optimize the performance of the model. Additionally, depth-wise separable convolutions are employed in the feature fusion process. The ALdamage-seg’s weight on the Alibaba Tian-chi aluminum profile dataset constitutes 43.9% of that of YOLOv8n-seg, with its GFLOPs reduced to 53% relative to YOLOv8-seg, all the while achieving an average precision (mAP) of 99% relative to YOLOv8-seg. With its compact size and lower computational requirements, this model is well-suited for deployment on edge devices with limited processing capabilities.
科研通智能强力驱动
Strongly Powered by AbleSci AI