User Profile

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!!