Only image inputs are supported for import using 'importONNXNetwork'
7 views (last 30 days)
My question in regarding the add-on 'importONNXNetwork'.
I am unable to load a ONNX model in MATLAB (2018b). The model is trained in PyTorch.
Input to the model is a vector and not an image.
I get following error: 'Only image inputs are supported for import'.
I am wondering if this is because MATLAB add-on doesn't (yet) support loading of ONNX models with non-image inputs.
Please let me know. Thanks in advance.
Josiah Wai on 17 Jan 2022
Edited: Josiah Wai on 17 Jan 2022
I also had this problem and was able to get around it by reshaping the onnx input vector.
- To export from pytorch to onnx, follow the steps at https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html However, before running the torch.onnx.export command, set the shape of your input vector x. In my case, each input sample vector has size 33, so I did: x = torch.randn(1,1,33, requires_grad=True)
- Import to Matlab
net = importONNXNetwork(modelfile, 'OutputLayerType', 'regression')
x = rand(33,1)
y = predict(net, x)