AGCA: An Adaptive Graph Channel Attention Module for Steel Surface Defect Detection

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.