How to import pretrained models with multiple outputs to MATLAB?

2 visualizaciones (últimos 30 días)
Amin Assadzadeh
Amin Assadzadeh el 20 de Oct. de 2019
Respondida: Shashank Gupta el 9 de Dic. de 2019
I'm trying to import pretrained detector networks (e.g. YOLOv3 and Faster R-CNN) in ONNX or Keras formats using importONNXNetwork and importKerasNetwork functions, but I'm getting the error that models with multiple outputs are not supported yet?
Is there a way around this?
Any help will be much appreciated!

Respuestas (1)

Shashank Gupta
Shashank Gupta el 9 de Dic. de 2019
Hi Amin,
The issue which you mentioned is a known issue and the concerned team is working on it.
One thing that can be tried although I haven’t done it personally. Before importing the model to MATLAB environment, try adding a dummy layer with a dummy weight with just one output and after importing, remove the added layer and get back your original model architecture. I am not sure weather this will work but still worth giving a shot.
Hope this helps.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by