xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed autoformat errors

This commit is contained in:
AnonymDevOSS 2025-11-06 23:00:33 +01:00
parent ceefe0bab2
commit bb87ab0a6e
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/detection/utils/test_iou_and_nms.py
View File

@ -6,7 +6,6 @@ import numpy as np
import pytest
from supervision.detection.utils.iou_and_nms import (
OverlapMetric,
_group_overlapping_boxes,
box_iou,
box_iou_batch,
@ -718,24 +717,22 @@ def test_box_iou_batch_consistency_with_box_iou():
for i, box_true in enumerate(boxes_true):
for j, box_detection in enumerate(boxes_detection):
single_result = box_iou(box_true, box_detection)
assert np.allclose(
batch_result[i, j], single_result, rtol=1e-5, atol=1e-5
)
assert np.allclose(batch_result[i, j], single_result, rtol=1e-5, atol=1e-5)
def test_box_iou_batch_with_mock_detections():
""" Test box_iou_batch with generated boxes and verify results are valid. """
"""Test box_iou_batch with generated boxes and verify results are valid."""
boxes_true = np.array(mock_boxes(10, seed=1), dtype=np.float32)
boxes_detection = np.array(mock_boxes(15, seed=2), dtype=np.float32)
result = box_iou_batch(boxes_true, boxes_detection)
assert result.shape == (10, 15)
assert np.all(result >= 0)
assert np.all(result <= 1.0)
# and symetric
# and symmetric
result_reversed = box_iou_batch(boxes_detection, boxes_true)
assert result_reversed.shape == (15, 10)
assert np.allclose(result.T, result_reversed, rtol=1e-5, atol=1e-5)