Searched refs:bbox_a (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | transforms.py | 29 def bbox_iou(bbox_a, bbox_b, offset=0): argument
50 if bbox_a.shape[1] < 4 or bbox_b.shape[1] < 4:
53 tl = np.maximum(bbox_a[:, None, :2], bbox_b[:, :2])
54 br = np.minimum(bbox_a[:, None, 2:4], bbox_b[:, 2:4])
57 area_a = np.prod(bbox_a[:, 2:4] - bbox_a[:, :2] + offset, axis=1)
|