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

YoloNAS-R #1983

Closed
wants to merge 151 commits into from
Closed

YoloNAS-R #1983

wants to merge 151 commits into from

Conversation

BloodAxe
Copy link
Contributor

@BloodAxe BloodAxe commented May 7, 2024

This is a draft PR

What in the box:

  1. Model arch, loss and metrics
  2. Dataset
  3. Basic transforms
  4. model.predict()
  5. Notebook to generate dota submissions
  6. Train recipe
  7. Export Notebook
  8. Export support (Using MatrixNMS)

What is missing

  1. Pretrained weights
  2. Docs

BloodAxe added 26 commits May 13, 2024 14:29
# Conflicts:
#	src/super_gradients/training/utils/distributed_training_utils.py
…#1994)

* Split processing and default preprocesisng params into separate files

* Put back method for BC support

* Fix imports

* Revert sampelr
…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
* OBB loss

* DOTA dataset

* Adding loss, postprocessing & visualization

* Visualization callback

* Visualization callback

* YoloNAS-R

* YoloNAS-R

* YoloNAS-R

* optimized_rboxes_nms

* optimized_rboxes_nms

* optimized_rboxes_nms

* optimized_rboxes_nms

* optimized_rboxes_nms

* Rename variables for better clarity

* Optimize loss weights

* Prepare data script

* Prepare data script

* Reduce batch size & topk

* Remove unused arg

* Remove unused arg

* Remove unused arg

* Remove unused arg

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* Increase topk

* RAdam

* RAdam

* RAdam

* CIou=False

* Disable check_points_inside_rboxes

* Disable check_points_inside_rboxes

* Disable check_points_inside_rboxes

* Tune weights

* set_anomaly_enabled

* set_anomaly_enabled

* Increase eps

* Multiply by scalar

* Multiply by scalar

* average_losses_in_ddp

* yolo_nas_r_balanced

* Limit angle

* Optimize image slicing

* Do not reduce losses, reduce only scores sum

* Do not reduce losses, reduce only scores sum

* Do not reduce losses, reduce only scores sum

* Clip grad

* Disable L1 components

* Disable L1 components

* Disable L1 components

* Disable L1 components

* predict() support for YoloNAS-R

* Comment saving of visualization

* yolo_nas_r_tzag

* Remove parameter

* Fixed newline

* Increase min confidence

* Added missing ReverseImageChannels

* yolo_nas_r_tzag_balanced

* Added OBB transforms

* Added OBB transforms

* get_dataset_preprocessing_params

* Added removal of small boxes during training

* Added removal of small boxes during training

* Added removal of invalid boxes

* Added augmentations

* Increase bs

* Ensure that poly_to_cxcywhr always return boxes with w > h

* Enabled anomaly detection

* Added eps

* Undo anomaly

* dota_yolo_nas_r_balanced_no_mixup

* Added logging of non-finite IoU results

* Added sanitize sample call after applying albumentations transforms

* Replaced atan2() with atan()  in CIoU loss because gt boxes can be zeros

* MOAR Augs

* MOAR Augs

* RAdam -> AdamW
Less augs

* dota_yolo_nas_r_balanced_pretrain

* Tune augs

* Fixed issue of logging wrong config

* Revert file

* Cleanup

* Added docs

* Remove non-existing params

* YoloNAS-R M&L variants

* Export support for YoloNAS R

* Temporary comment matrix nms

* PrefetchIterator

* Remove gs import

* dota_yolo_nas_r_s

* Increase BS

* Enable anomaly detection

* Rewrite t3

* Rewrite t3

* Rewrite t3

* dota_yolo_nas_r_s_1_gpu

* M & L

* Increase numerical stability

* Remove prefetch

* Increase BS for L

* Increase BS for L

* Disable fp16

* RM anomaly

* RM anomaly

* fp32

* Increase batch size

* Increase batch size

* Increase batch size

* dota_yolo_nas_r_l_fp32

* Switch to use of matrix nms in post-prediction callback and exact polygon-polygon iou for metric

* Switch to use of matrix nms in post-prediction callback and exact polygon-polygon iou for metric

* Added docs for rboxes_matrix_nms

* Update script

* Improve auto-generated submission name

* Added positional args

* Added positional args

* dota_yolo_nas_r_l_fp32_mixup

* dota_yolo_nas_r_l_fp32_mixup

* dota_yolo_nas_r_l_fp32_mixup

* dota_yolo_nas_r_l_fp32_mixup

* Update recipes

* Update recipes

* Added license support for YoloNAS-R

* Improve docstrings

* Remove max gradient debugging

* Improve docstrings

* Remove random rotate90

* Move OBBSample to samples

* Added test
…feature/SG-1448-OBB

# Conflicts:
#	src/super_gradients/training/datasets/data_formats/obb/cxcywhr.py
@BloodAxe BloodAxe marked this pull request as ready for review June 4, 2024 07:14
@BloodAxe BloodAxe closed this Jun 4, 2024
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.

1 participant