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

Feature/sg 1488 yolo nas r integration sample and transforms #1995

Conversation

BloodAxe
Copy link
Contributor

@BloodAxe BloodAxe commented May 17, 2024

Adds OBBSample dataclass and transforms related to OBB sample.
Plus Albu support

Copy link
Contributor

@shaydeci shaydeci left a comment

Choose a reason for hiding this comment

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

Some short minor comments.
The use of keypoints for rotated boxes for the Albu support is awesome! Lets just add another use case for testing in the albumentations integration test like we have for the other tasks (tests/integration_tests/albumentations_test.py).
Also, pretty sure I found some problems with the rotate90 transform, so just to be safe it might be a good idea to have just one test that forward some ample through all of the newly added transforms.

Copy link
Contributor

@shaydeci shaydeci left a comment

Choose a reason for hiding this comment

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

LGTM

@BloodAxe BloodAxe merged commit 3013463 into feature/SG-1488-YoloNAS-R-integration May 21, 2024
3 of 4 checks passed
@BloodAxe BloodAxe deleted the feature/SG-1488-YoloNAS-R-integration-sample-and-transforms branch May 21, 2024 13:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants