BatchMulti​ClassNonMa​xSuppressi​on - importONNXLayers/Network issue

1 次查看（过去 30 天）
Vincent Dufour
Vincent Dufour 2020-2-6
评论： Veer Bhansari 2020-6-11
Hi guys,
I'm currently trying to do some DeepLearning using Matlab. I saw it was possible to import ONNX model directly in Matlab for training.
So I tried to import ssd_mobilenet_v1_fpn_coco from this official model zoo tensorflow repository:
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
After I convert it to ONNX format (using tf2onnx) I tried to import it using "importONNXNetwork" and "importONNXLayers" (with classification as the OutputLayers). Here are the errors I get from Matlab:
The datatype of initializer 'Postprocessor/BatchMultiClassNonMaxSuppression/map/while/PadOrClipBoxList/cond/cond/pred_id/_174__cf__177:0'
('BOOL') is not supported yet.
From what I understand it seems that Matlab doesn't support this type of layers, so my question is: what can I do to solve this ? Is there a way to implement this layer or to tell matlab to delete it and replace by another ?
Thank you in advance

请先登录，再回答此问题。

采纳的回答

Divya Gaddipati
Divya Gaddipati 2020-2-10
For building a custom layer, please refer to the below link:
https://www.mathworks.com/help/deeplearning/ug/define-custom-deep-learning-layers.html
And you can use the function replaceLayer to replace the layer with your custom layer
For more information on how to use replaceLayer, refer to the following link:
https://www.mathworks.com/help/deeplearning/ref/replacelayer.html
Hope this helps!
  6 个评论
显示 4更早的评论
Vincent Dufour
Vincent Dufour 2020-6-11
Hey,
nope I totally skipped Matlab, at first I found their feature to manage the data really cool (especially the video labelling feature) but then the problem with certain layers stopped me.
I used retinanet-examples repo from Nvidia on Github.
Cheers
Veer Bhansari
Veer Bhansari 2020-6-11
Hi,
Yes, the Image and Video labeller tools are handy and even I wanted to explore them. But it seems, we can't leverage them completely as of now due to these limitations.
Thanks!

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by