Skip to content

Fixed conversion of the TF OD API 2.0 SSD models#4529

Merged
lazarevevgeny merged 1 commit intoopenvinotoolkit:masterfrom
lazarevevgeny:fix_tf_od_api_2_ssd_models
Mar 1, 2021
Merged

Fixed conversion of the TF OD API 2.0 SSD models#4529
lazarevevgeny merged 1 commit intoopenvinotoolkit:masterfrom
lazarevevgeny:fix_tf_od_api_2_ssd_models

Conversation

@lazarevevgeny
Copy link
Copy Markdown
Contributor

@lazarevevgeny lazarevevgeny commented Mar 1, 2021

Description: Fixed conversion of TF OD API 2.X SSD models. The problem appeared after #4367 where the StridedSlice operation transformation was changed and ShapeOf operations appeared in the sub-graph being added. The TF OD API models conversion relies on the fact that the part of sub-graph with prior boxes must be const folded and the data node with value of prior boxes is modified in the transformation to align from the TF semantic to OpenVINO semantic. Because of the change in #4367 the sub-graph is not const folded and the change of the data node of prior boxes is ignored. The fix is to create a Const node explicitly so the const sub-graph is removed.

Ticket: 50020

Code:

  • Comments
  • Code style (PEP8)
  • Transformation generates reshape-able IR
  • Transformation preserves original framework node names

Validation:

  • Unit tests: N/A for such a change
  • Framework operation tests: N/A no new operation added
  • Transformation tests: N/A for such a change. Will be tested by the e2e tests
  • Model Optimizer IR Reader check: Done

Documentation:

  • Supported frameworks operations list: N/A no new operation enabled
  • Guide on how to convert the public model: N/A no user visible changes
  • User guide update: N/A no user visible changes

@lazarevevgeny lazarevevgeny marked this pull request as ready for review March 1, 2021 06:08
@lazarevevgeny lazarevevgeny requested review from a team, achetver, vgavrilo and yekruglov and removed request for a team March 1, 2021 06:08
@openvino-pushbot openvino-pushbot added the category: MO Model Optimizer label Mar 1, 2021
Copy link
Copy Markdown
Contributor

@achetver achetver left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me.

@lazarevevgeny lazarevevgeny merged commit 62fb80a into openvinotoolkit:master Mar 1, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

category: MO Model Optimizer

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants