You’re welcome. I’ve actually been working on a segmentation problem myself and had the same issue 
. Feel free to ask again if you need more help.
Also, you may find this conversation relevant: https://forums.fast.ai/t/unet-binary-segmentation/29833/8