No, unfortunately I didn’t create such a nice figure for this implementation.
Yes, I used UNet as the base model and have rewritten it a bit to work with differently shaped inputs and also to output the same spatial size (instead of a cropped version like in the original implementation).