I train some Unet- based model in Pytorch. It take an image as an input, and return a mask. After training i convert model to ONNX format, run it with onnxruntime python mod