- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
I'm using openvino version 2021 2021.4 and tensorflow version 1.15.5 in my virtual env.
The model i was trying to convert is "HEAD_DETECTION_300x300_ssd_mobilenetv2.pb", which I'm attaching a link here to download the model:
https://github.com/Varat7v2/Human-Head-Detection/tree/master/models
I'm using this conversion command:
-------------------------------------------
python3 /opt/intel/openvino_2021/deployment_tools/model_optimizer/mo_tf.py \
--input_model=HEAD_DETECTION_300x300_ssd_mobilenetv2.pb \
--tensorflow_use_custom_operations_config ssd_v2_support.json \
--tensorflow_object_detection_api_pipeline_config pipeline.config \
--batch 1 \
--reverse_input_channels
-------------------------------------------
However, i'm getting error
[ ERROR ] Cannot infer shapes or values for node "Postprocessor/ToFloat".
Then i chnaged the "Postprocessor/Cast_1" to "Postprocessor/ToFloat" inside ssd_v2_support.json
However, i'm now getting a different error:
-------------------------------------------
[ ERROR ] Cannot infer shapes or values for node "BoxPredictor_0/ClassPredictor/BiasAdd/Reshape".
[ ERROR ] Number of elements in input [ 1 19 19 6] and output [1, 23, 91] of reshape node BoxPredictor_0/ClassPredictor/BiasAdd/Reshape mismatch
-------------------------------------------
Can i have some help regarding this problem please? Thank you very much.
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Solved the issue by editing "pipeline.config" to have:
num_classes: 1
Thank you
Link Copied
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Solved the issue by editing "pipeline.config" to have:
num_classes: 1
Thank you
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Hi abhi_jci,
Thank you for sharing your solution with our Community. I'm glad to hear that. Hope everything is going well for you. This thread will no longer be monitored since this issue has been resolved. If you need any additional information from Intel, please submit a new question.
Regards,
Megat

- Subscribe to RSS Feed
- Mark Topic as New
- Mark Topic as Read
- Float this Topic for Current User
- Bookmark
- Subscribe
- Printer Friendly Page