Research Article

Casting Defect Detection and Classification of Convolutional Neural Network Based on Recursive Attention Model

Table 4

The impact of different modules on network performance.

ItemFLOPsXLDDefect detectionImage processingPyTorchResNet-50

ShuffleNetV23.024.815.425.782.46
Mix-fusion1.352.933.315.395.012.69
Loss4.734.31.983.323.823.98
ShuffleNet1.51.31.243.191.766.82
Fourier2.415.523.095.845.684.72
Casting1.986.391.284.756.672.84