Research Article

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

Table 1

The number of parameters of the compared attention mechanism.

ModelParameters

3D-UNet [27]1459585
3D-UNet-SE [34]1634865
3D-UNet-CBAM [36]1465515
3D-UNet-ECA [35]1557873