Audit and suppress nested detector proposals
This commit is contained in:
@@ -26,3 +26,15 @@ def test_empty_reference_counts_false_positives() -> None:
|
||||
|
||||
def test_box_scaling_preserves_center() -> None:
|
||||
assert MODULE.scale_box((10.0, 20.0, 30.0, 40.0), 1.5) == (5.0, 15.0, 35.0, 45.0)
|
||||
|
||||
|
||||
def test_containment_suppression_removes_nested_lower_score_box() -> None:
|
||||
predictions = [
|
||||
((0.0, 0.0, 20.0, 20.0), 0.9),
|
||||
((5.0, 5.0, 15.0, 15.0), 0.8),
|
||||
((25.0, 0.0, 35.0, 10.0), 0.7),
|
||||
]
|
||||
assert MODULE.suppress_contained_predictions(predictions, 0.8) == [
|
||||
predictions[0],
|
||||
predictions[2],
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user