Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add OV model for keypoint detection #3970

Merged
merged 2 commits into from
Sep 26, 2024

Conversation

sovrasov
Copy link
Contributor

Summary

  • OVModel
  • fix torch functionality (export & benchmark)
  • minor fix in metric computation
  • CLI&Unit tests

Results on coco_keypoint_single_obj/medium:

Model PCK accuracy
RTMPose-tiny-single 0.7216
RTMPose-tiny-single OV 0.7216
RTMPose-tiny-single OV INT8 0.7208

previous PR: #3957

@sovrasov
Copy link
Contributor Author

sovrasov commented Sep 20, 2024

Waiting for #3963 [merged]

Add integration tests for keypoint detection task (openvinotoolkit#3945)

* Fix export

* Add integrastion tests for keypoint detection

* Update export

* Add kp det to gh workflow

* Fix ruff

* Move std mean to class vars

Cleanup

Implement dummy input for kp

Update export inf test

Upgrade MAPI

Cleanup

Fix CLI tests

Cleanup

Fix kp OV model data config

Fix dummy input docstring

Co-authored-by: Prokofiev Kirill <kirill.prokofiev@intel.com>

Add unit tests for kp det model

Fix ruff

Update pck docs

Fix typos in tests

Add uts for PCK metric

Rename kaypoints det metric

Del extra transform in p ove model

Fix unit test naming

Fix cpu train/eval

Fix autocast deprecation warnings
@sovrasov sovrasov marked this pull request as ready for review September 25, 2024 11:35
@eunwoosh eunwoosh enabled auto-merge September 26, 2024 00:39
@eunwoosh eunwoosh added this pull request to the merge queue Sep 26, 2024
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Sep 26, 2024
@sovrasov
Copy link
Contributor Author

tox -vv -e integration-test-detection
image

Detection tests passed locally. The branch is up-to-date with develop. Merge queue is not functional because of random failures.

@sovrasov sovrasov merged commit 063788a into openvinotoolkit:develop Sep 26, 2024
18 of 22 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
TEST Any changes in tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants