Thanks for the update. Your use case sounds indeed like using images with black and while pixel values, so you should be able to transform it to float32 in the input.
I’m currently still unsure how the output would be treated, i.e. are you using the bool output to classify each pixel location or to restore the original input? In that case also the output could be converted to float32.