Research Article

Fault Recognition Method Based on Attention Mechanism and the 3D-UNet

Table 2

The number of parameters and FLOPs of the compared segmentation model.

ModelParametersGFLOPs

3D-UNet-SE [34]16348652.72e + 02 G
FCN4 [37]150333721.7 G
FCN8 [37]161568921.6 G
AttUNet [38]17560083.76e + 02 G