diff --git a/test/metrics/test_detection.py b/test/metrics/test_detection.py index a25c7857..fe51630a 100644 --- a/test/metrics/test_detection.py +++ b/test/metrics/test_detection.py @@ -95,10 +95,10 @@ for det in DUMMY_DET_DATASET.annotations.values(): "predictions, targets, classes, conf_threshold, iou_threshold, expected_result, exception", [ ( - [CERTAIN_DETECTIONS], [DETECTIONS], + [CERTAIN_DETECTIONS], CLASSES, - 0.3, + 0.2, 0.5, IDEAL_CONF_MATRIX, DoesNotRaise(), @@ -134,7 +134,7 @@ def test_from_detections( DETECTION_TENSORS, CERTAIN_DETECTION_TENSORS, CLASSES, - 0.3, + 0.2, 0.5, IDEAL_CONF_MATRIX, DoesNotRaise(), @@ -170,7 +170,7 @@ def test_from_tensors( DETECTION_TENSORS[0], CERTAIN_DETECTION_TENSORS[0], NUM_CLASSES, - 0.3, + 0.2, 0.5, IDEAL_CONF_MATRIX, DoesNotRaise(),