Deformation Feature Extraction and Double Attention Feature Pyramid Network for Bearing Surface Defects Detection

被引:5
|
作者
Peng, Yongkang [1 ]
Xia, Fei [1 ]
Zhang, Chuanlin [1 ]
Mao, Jianliang [1 ]
机构
[1] Shanghai Univ Elect Power, Intelligent Autonomous Syst Lab, Shanghai 200090, Peoples R China
基金
中国国家自然科学基金;
关键词
Feature extraction; Convolution; Shape; Residual neural networks; Defect detection; Surface treatment; Convolutional neural networks; Attention mechanism; deformable convolution; feature pyramid fusion; surface defect detection; INSPECTION;
D O I
10.1109/TII.2024.3370330
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In vision-based bearing defect detection, random shapes and multiscale defects can result in inaccurate feature extraction, and complex backgrounds can interfere with defect localization. Therefore, a novel bearing defect detection network (BDDN) is proposed in this article, aiming to improve the accuracy of bearing defect detection, which consists of deformable ResNet50 network (DRN) and double attention feature pyramid network (DA-FPN). In the DRN, it is found that using deformable convolution for stages 1-3 of ResNet50 is able to extract features that are closer to the actual defect regions through rich experiments. The DA-FPN is employed to reduce the interference of complex backgrounds for defect detection, which consists of a feature-enhanced attention module (FEAM) and a feature-communication attention module (FCAM). The FEAM obtains more refined high-level semantic information in channel feature extraction, while the FCAM efficiently fuses defect features across different levels. Experimental results demonstrate that the BDDN effectively detects bearing cover surface defects, achieving a mean average precision of 87.9%. In addition, it outperforms several defect detection models on the public Northeastern University dataset (NEU-DET) in terms of accuracy.
引用
收藏
页码:9048 / 9058
页数:11
相关论文
共 50 条
  • [21] Feature enhancement modules applied to a feature pyramid network for object detection
    Liu, Min
    Lin, Kun
    Huo, Wujie
    Hu, Lanlan
    He, Zhizi
    PATTERN ANALYSIS AND APPLICATIONS, 2023, 26 (02) : 617 - 629
  • [22] Integrated Feature Pyramid Network With Feature Aggregation for Traffic Sign Detection
    Tang, Qing
    Cao, Ge
    Jo, Kang-Hyun
    IEEE ACCESS, 2021, 9 : 117784 - 117794
  • [23] Feature enhancement modules applied to a feature pyramid network for object detection
    Min Liu
    Kun Lin
    Wujie Huo
    Lanlan Hu
    Zhizi He
    Pattern Analysis and Applications, 2023, 26 : 617 - 629
  • [24] Multilevel Pyramid Feature Extraction and Task Decoupling Network for SAR Ship Detection
    Li, Yanshan
    Liu, Wenjun
    Qi, Ruo
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 3560 - 3570
  • [25] An improved feature pyramid network for object detection
    Zhu, Linxiang
    Lee, Feifei
    Cai, Jiawei
    Yu, Hongliu
    Chen, Qiu
    NEUROCOMPUTING, 2022, 483 : 127 - 139
  • [26] Parallel Feature Pyramid Network for Object Detection
    Kim, Seung-Wook
    Kook, Hyong-Keun
    Sun, Jee-Young
    Kang, Mun-Cheon
    Ko, Sung-Jea
    COMPUTER VISION - ECCV 2018, PT V, 2018, 11209 : 239 - 256
  • [27] Complementary Feature Pyramid Network for Object Detection
    Xie, Jin
    Pang, Yanwei
    Pan, Jing
    Nie, Jing
    Cao, Jiale
    Han, Jungong
    ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2023, 19 (06)
  • [28] Latent Feature Pyramid Network for Object Detection
    Xie, Jin
    Pang, Yanwei
    Nie, Jing
    Cao, Jiale
    Han, Jungong
    IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 : 2153 - 2163
  • [29] Gated Feature Pyramid Network for Object Detection
    Xie, Xuemei
    Liao, Quan
    Ma, Lihua
    Jin, Xing
    PATTERN RECOGNITION AND COMPUTER VISION (PRCV 2018), PT IV, 2018, 11259 : 199 - 208
  • [30] Feature Extraction of Defects on Wood Surface
    王克奇
    白景峰
    莫虹
    孔详林
    崔克冰
    Journal of Forestry Research, 1998, (01) : 54 - 56