I downloaded the mobilenetv2 model from deeplab model zoo :
https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/...
And when I run the command
"snpe-tensorflow-to-dlc --graph mobilenetv2_deeplab/frozen_inference_graph.pb --input_dim sub_7 513,513,3 --dlc mobilenetv2_deeplab.dlc --out logits/semantic/BiasAdd"
I got the erre like below:
layers/eltwise.py:164: RuntimeWarning: error_code=1004; error_message=Layer parameters combination is invalid. Layer MobilenetV2/expanded_conv_7/add: dimensions mismatch between MobilenetV2/expanded_conv_6/project/BatchNorm/FusedBatchNorm:0 {65, 65, 64} and MobilenetV2/expanded_conv_7/add {61, 61, 64}; error_component=Model Validation; line_no=117; thread_id=140619208206144
But I printed the shapes of inputs of "MobilenetV2/expanded_conv_7/add", I got [1, 65, 65, 64] and [1, 65, 65, 64].
Someone can help me?
Hi Fang,
The input dimensions provided for arguments are incorrect.
Please find the below command for right usage,
$ snpe-tensorflow-to-dlc --graph mobilenetv2_deeplab/frozen_inference_graph.pb --input_dim sub_7 1,513,513,3 --out_node ArgMax --dlc deeplabv3.dlc --allow_unconsumed_nodes
Hope this helps.