阶段(地层学)
纹理(宇宙学)
特征(语言学)
模式识别(心理学)
人工智能
网(多面体)
计算机科学
材料科学
数学
图像(数学)
地质学
几何学
古生物学
语言学
哲学
作者
Hao Zhong,Dongdong Fu,Xiao Ling,Fang Zhao,Jie Liu,Youmin Hu,Bo Wu
标识
DOI:10.1016/j.aei.2024.102437
摘要
Statistical texture features are essential for metal surface defect detection. However, low contrast and cluttered backgrounds exacerbate the intrinsic blurriness and variability of defect textures, hampering accurate identification and localization. To tackle these challenges, we propose a Statistical Texture Feature Enhancement Network (STFE-Net) based on YOLO model, a multi-stage approach based on pooling attention mechanism and cross-scale shallow feature reinforcement. Firstly, we enhance the image contrast and propose a Stacked Pooling Attention Residual (SPA-Res) module to capture better texture features; then, we embed the Shallow Feature Weighted (SFW) branches to achieve cross-layer shallow feature reinforcement and add feature layer (AFL) to extract rich semantic features; and finally, we propose an Efficient Complete Intersection over Union (ECIoU) loss function and employ four detection heads to achieve precise localization of defects. On the NEU-DET dataset, our model achieves the state-of-the-art performance. Especially, the mean Average Precision (mAP@IoU = 0.5) of our model is 82.3 %, which is 6.2 % higher than baseline network. With the input resolution of 640 × 640, STFE-Net achieves 54 FPS inference speed with a GeForce GTX 1080. In addition, the experimental results on the DAGM dataset and GC10-DET dataset demonstrate that the proposed model has excellent generalization ability.
科研通智能强力驱动
Strongly Powered by AbleSci AI