Grayscale images in u-net

I am far from an expert but I am using UNetMini from this example. This is a post with some of my code where I got some help not to long ago. Might find it somewhat useful.