I trained a UNet model using Pytorch and used coremltools to convert it to the CoreML format.
I found inconsistent results after the conversion. The CoreML model predicts all black masks while the results of pytorch model look fine. I have confirmed that there is no preprocessing done as part of the model training/trace process.
Can someone help with this please?
Thank you!!
