-
Notifications
You must be signed in to change notification settings - Fork 510
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
Feature/sg 1488 yolo nas r integration sample and transforms #1995
Conversation
…ygon-polygon iou for metric
…ygon-polygon iou for metric
# Conflicts: # src/super_gradients/training/utils/distributed_training_utils.py
…feature/SG-1448-OBB # Conflicts: # src/super_gradients/training/processing/__init__.py # src/super_gradients/training/processing/defaults.py # src/super_gradients/training/processing/processing.py
There was a problem hiding this 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.
src/super_gradients/training/transforms/obb/obb_random_rotate90.py
Outdated
Show resolved
Hide resolved
src/super_gradients/training/transforms/obb/obb_random_rotate90.py
Outdated
Show resolved
Hide resolved
src/super_gradients/training/transforms/obb/obb_random_rotate90.py
Outdated
Show resolved
Hide resolved
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
3013463
into
feature/SG-1488-YoloNAS-R-integration
Adds OBBSample dataclass and transforms related to OBB sample.
Plus Albu support