…#1994)

* Split processing and default preprocesisng params into separate files

* Put back method for BC support

* Fix imports

* Revert sampelr

* 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

* Added DOTA2 dataset

* Added DOTA2 dataset

* Add missing type

* Added dataset setup instructions

* Added extreme batch visualization callback

* OBBDetectionMetrics inherit from DetectionMetrics

* Added AbstractOBBDataset

* Added docs to prepare dataset script

* Update docs on visualization callback

* Update docs on dataset config file

* Added DOTA2 dataset

* Added DOTA2 dataset

* Add missing type

* Added dataset setup instructions

* Added extreme batch visualization callback

* OBBDetectionMetrics inherit from DetectionMetrics

* Added AbstractOBBDataset

* Added docs to prepare dataset script

* Update docs on visualization callback

* Update docs on dataset config file

* Model, loss and recipes

* Addedd export notebook

* Added export support

* Added predict script for generation a submission to DOTA test-dev

* Added missing imports

* Make YoloNASRDFLHead inherit a base YoloNASDFLHead

* Make YoloNASRDFLHead inherit a base YoloNASDFLHead

* Added explicit mention of TRT support

* Added explicit mention of TRT support

* Added new recipes to test

* Added test to ensure flat_collate_tensors_with_batch_index works correct with bool tensors

* Move limitations to the top

* Added inheritance