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 条
  • [1] Pyramid Feature Attention Network for Saliency detection
    Zhao, Ting
    Wu, Xiangqian
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 3080 - 3089
  • [2] Pyramid Feature Attention Network for Speech Resampling Detection
    Zhou, Xinyu
    Zhang, Yujin
    Wang, Yongqi
    Tian, Jin
    Xu, Shaolun
    APPLIED SCIENCES-BASEL, 2024, 14 (11):
  • [3] Detection of PCB Surface Defects With Improved Faster-RCNN and Feature Pyramid Network
    Hu, Bing
    Wang, Jianhui
    IEEE ACCESS, 2020, 8 : 108335 - 108345
  • [4] Dual Attention Based Feature Pyramid Network
    Xing, Huijun
    Wang, Shuai
    Zheng, Dezhi
    Zhao, Xiaotong
    CHINA COMMUNICATIONS, 2020, 17 (08) : 242 - 252
  • [5] Method of feature pyramid and attention enhancement network for pavement crack detection
    Qu, Zhong
    Dong, Xiao-Yu
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (03)
  • [6] A lightweight vehicle detection network fusing feature pyramid and channel attention
    Lu, Jian
    Huang, Tingting
    Zhang, Qi
    Chen, Xiaogai
    Zhou, Jian
    INTERNET OF THINGS, 2024, 26
  • [7] Dual Attention Based Feature Pyramid Network
    Huijun Xing
    Shuai Wang
    Dezhi Zheng
    Xiaotong Zhao
    中国通信, 2020, 17 (08) : 242 - 252
  • [8] Joint Attention-Guided Feature Fusion Network for Saliency Detection of Surface Defects
    Jiang, Xiaoheng
    Yan, Feng
    Lu, Yang
    Wang, Ke
    Guo, Shuai
    Zhang, Tianzhu
    Pang, Yanwei
    Niu, Jianwei
    Xu, Mingliang
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2022, 71
  • [9] Bidirectional Feature Pyramid Network with Recurrent Attention Residual Modules for Shadow Detection
    Zhu, Lei
    Deng, Zijun
    Hu, Xiaowei
    Fu, Chi-Wing
    Xu, Xuemiao
    Qin, Jing
    Heng, Pheng-Ann
    COMPUTER VISION - ECCV 2018, PT VI, 2018, 11210 : 122 - 137
  • [10] AFFPN: Attention Fusion Feature Pyramid Network for Small Infrared Target Detection
    Zuo, Zhen
    Tong, Xiaozhong
    Wei, Junyu
    Su, Shaojing
    Wu, Peng
    Guo, Runze
    Sun, Bei
    REMOTE SENSING, 2022, 14 (14)