mirror of
https://github.com/VikParuchuri/surya.git
synced 2026-06-04 21:03:53 +08:00
9 lines
286 B
Python
9 lines
286 B
Python
def test_detection(detection_predictor, test_image):
|
|
detection_results = detection_predictor([test_image])
|
|
|
|
assert len(detection_results) == 1
|
|
assert detection_results[0].image_bbox == [0, 0, 1024, 1024]
|
|
|
|
bboxes = detection_results[0].bboxes
|
|
assert len(bboxes) == 4
|