please point to any example code implementation of tensorflow 3D Unet to gray scale image mask . Thank you