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

Figure 9

The recognition effect of different attention mechanism models. (a) 3D-UNet + SE model prediction results. (b) 3D-UNet + CBAM model prediction results. (c) 3D-UNet model prediction results. (d) 3D-UNet + ECA model prediction results.

(a)

(b)

(c)

(d)