Shailendra-IISc

I converted the mask images from 32 bit to 8 bit. Then, the code runs successfully.

I used Paint.NET software to save the mask image as 8-bit type. Please make sure that the background is black and the mask is white.

Thank you, Bastian sir for the help. I am using Tensorflow 2.8.2 in Google CoLab GPUs. It is working now after implementing the suggested corrections. Regards, Shailendra