pytorch, onnx, and mlmodel

I've trained my GAN pytorch model, converted into onnx, and then mlmodel.


but the mlmodel's input/output type is MLMultiarray.


How can I change this into Image?


Which stage (pytorch or onnx or mlmodel) should I retouch?