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