def od_sample_detection_bboxes(): return [ DetectionBbox.from_array( [109.0, 190.0, 205.0, 408.0], label_idx=3, label_name="carton", score=0.9985, ), DetectionBbox.from_array( [340.0, 326.0, 465.0, 549.0], label_idx=2, label_name="milk_bottle", score=0.9979, ), DetectionBbox.from_array( [214.0, 181.0, 315.0, 460.0], label_idx=1, label_name="can", score=0.9945, ), DetectionBbox.from_array( [215.0, 193.0, 316.0, 471.0], label_idx=2, label_name="milk_bottle", score=0.1470, ), DetectionBbox.from_array( [109.0, 209.0, 209.0, 420.0], label_idx=1, label_name="can", score=0.0903, ), ]
def test_detection_bbox_from_array(det_bbox): bbox_from_array = DetectionBbox.from_array([0, 10, 100, 1000], label_idx=0, score=0) validate_bbox(det_bbox) assert type(bbox_from_array) == DetectionBbox