AFFNet: Adaptive feature fusion network for defect detection of industrial product surface
摘要
Industrial product surface defect detection plays a crucial role in guaranteeing the quality of industrial products. Several primary challenges remain in achieving efficient and accurate automatic detection: (1) Industrial product surface defects demonstrate irregular morphological features. (2) Significant size disparities between different defects result in the risk of information loss during feature fusion in Feature Pyramid Networks (FPN) typically employed in the neck component. (3) Visual similarity among different defect categories means that slight deviations in predicted box locations can result in misclassification errors. This research initially employs a DCN-C3k2 module design, incorporating DCNv3 to improve the model’s sensitivity to varied morphological information of objects. Furthermore, we design an Adaptive Focusing Diffusion Network (AFDN) that aggregates multi-scale features and implements adaptive channel selection and fusion, then propagates critical features through upsampling and downsampling processes to improve defect detection precision. Lastly, we introduce a Task Dynamic Interactive Detection Head (TDIDH). The TDIDH constructs detection head architecture according to the specific properties and variations of different detection tasks, with the objective of optimizing detection performance through enhanced inter-task dynamic interaction. Experiments are performed on public datasets GC10-DET and NEU-DET. The proposed method achieves AP, AP