openvinotoolkit / model_api

Apache License 2.0

25 stars 16 forks source link

Don't rely on outputs order from adapter in cls wrapper #165

Closed sovrasov closed 7 months ago

sovrasov commented 7 months ago

What does this PR do?

OVMS adapter can return the names in an arbitrary order, while the cls wrapper expects logits to be at the first place.

Fixes # (issue)

Before submitting

[ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[ ] Did you make sure to update the documentation with your changes?
[ ] Did you write any new necessary tests?

sovrasov commented 7 months ago

@RHeckerIntel please have a look