Only image inputs are supported for import using 'importONNXNetwork'
显示 更早的评论
Hello,
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.
Regards,
Pramod
回答(3 个)
matteo baigura
2020-2-1
编辑:matteo baigura
2020-2-1
2 个投票
Same problem, if you are right that's simply ridicolous
Josiah Wai
2022-1-17
编辑:Josiah Wai
2022-1-17
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)
yanqi liu
2022-1-17
0 个投票
yes,sir,may be transfer data to cell and save as mat file,then use reshape to modify data dimension
类别
在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
产品
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!