Visualization of the reconstruction pipeline and our NN architecture. The model consists of a U-net architecture with a nonlinear encoder and a decoder with skip connections. The nonlinear part uses leaky ReLU activation functions. Connections in the encoder part of the network use convolutional layers with a stride of 2, while the decoder part consists of transpose convolutional layers to ensure matching dimensions.