返回
Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection
DOI:10.1109/TIM.2024.3372229.png)
摘要
En 中文
Deep-learning-based detection methods have been widely applied to industrial defect inspection. However, directly using vanilla detection methods fails to achieve satisfying performance due to the lack of identifiable features. In this article, a novel attention-based multiscale feature fusion module (AMFF) is proposed, aiming to enhance defect features and improve defect identification by leveraging attention mechanism in the feature fusion. AMFF includes self-enhanced attention module (SEAM) and cross-enhanced attention module (CEAM). SEAM is performed on a single feature map, which first adopts multiple dilation convolutions to enrich contextual information without compromising resolution and then utilizes the intralayer attention on the current feature map. CEAM takes both the current feature map and the adjacent feature map as input to perform cross-layer attention. The adjacent feature map is modulated with the guidance of the current feature map, which is then combined with the current feature map and the output of SEAM for final prediction. AMFF is utilized in current feature fusion networks, e.g., feature pyramid network (FPN) and path aggregation FPN (PAFPN), and is further integrated into prevalent detectors to guide them to pay more attention to defects rather than the background. Extensive experiments are conducted on two real industrial datasets released by Tianchi platform, i.e., fabric and aluminum defect datasets. For each dataset, 500 images are randomly selected for test and the rest for training. The proposed AMFF is demonstrated to significantly boost defect detection accuracy with acceptable computational cost, and the real-time performance could fully satisfy practical requirements.
Keyword:
Attention mechanism
feature enhancement
feature fusion
object detection
surface defect detection
期刊
IF:
5.9
论文数:
2.0W
被引数:
5.8W
机构
引用论文
Surface Defects Detection and Identification of Lithium Battery Pole Piece Based on Multi-Feature Fusion and PSO-SVM基于多特征融合和PSO-SVM的锂电池极片表面缺陷检测与识别
IEEE ACCESS
IF3.6
Thermal comfort, perceived air quality, and cognitive performance when personally controlled air movement is used by tropically acclimatized persons当热带适应的人使用个人控制的空气运动时,热舒适性,感知的空气质量和认知表现
Indoor Air
IF0
Do Perceptions of Competence Mediate The Relationship Between Fundamental Motor Skill Proficiency and Physical Activity Levels of Children in Kindergarten?能力的感知是否可以介导幼儿园儿童的基本运动技能熟练程度与身体活动水平之间的关系?
Real-Time Defect Detection of Track Components: Considering Class Imbalance and Subtle Difference Between Classes轨道部件的实时缺陷检测: 考虑类别不平衡和课程之间的细微差异

