Matthew O'Malley-Nichols 14db029494 Soft Non-Maximum Suppression (#2400) ...

* Soft NMS with thresholds

* NMS Test

* Soft nms w/ boxes removed below threshold

* Soft nms test

* No longer removing bounding boxes to fit Soft-NMS focus

* Initialize confidence

* Added comments

* Refactored out updating based on IOU/sigma

* Score_threshold -> confidence_threshold for clarity

* Remove bboxes below confidence threshold

* Softnms basic functionality test

* Softnms confidence decay test

* Softnms confidence threshold test

* Softnms no overlapping bbox test

* Testing confidence after no overlap test

* Single bbox and no bbox tests

* Signify test completion

* Handling result of test functions

* Checking all pairs of bboxes instead of a forward pass

* Equal confidence overlap test

* Clarified tests for implementation

* No longer dropping boxes, just setting to 0.0

* Formatted w/ cargo

2024-08-10 07:57:52 +02:00

..

generation_tests.rs

Add topk sampling. (#1923)

2024-03-23 15:26:09 +01:00

nms_tests.rs

Soft Non-Maximum Suppression (#2400)

2024-08-10 07:57:52 +02:00