Abstract
Surface defect detection is an important part of the steel production process. Recently, attention mechanisms have been widely used in steel surface defect detection to ensure product quality. Existing attention modules cannot distinguish the difference between steel surface images and natural images. Therefore, we propose an Adaptive Graph Channel Attention (AGCA) module which introduces graph convolutional theory into channel attention. The AGCA module takes each channel as a feature vertex, and their relationship is represented by an adjacency matrix. We perform non-local operations on features by analyzing graphs constructed in AGCA. The operation significantly improves the feature representation capability. Like other attention modules, AGCA has lightweight and plug-and-play characteristics. It enables the module easily embedded into defect detection networks. The experimental results on various backbone networks and datasets show that AGCA outperforms state-of-the-art methods. Code is available at https://github.com/C1nDeRainBo0M/AGCA.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have