Hi, thanks for this great work! I wonder if you could share the decoder structure for decoding the output from the U-Net middle block into 6 planes?