Fig. 2

Overview of the electron density pipeline. The modification block consists of 1 × 1 convolution followed by adaptive max pooling, batch normalization and two fully connected layers. The resulting segmentation cube from the UNet is multiplied and passed through the modified block. The fully connected layer assigns class probabilities for the given input using the provided class labels