项目场景:在faster-rcnn求iou遇到
tl = np.maximum(bbox_a[:, None, :2], bbox_b[:, :2]) br = np.minimum(bbox_a[:, None, 2:], bbox_b[:, 2:]) area_i = np.prod(br - tl, axis=2) * (tl < br).all(axis=2)
问题描述:
python 广播机制–>避免用循环
列表a 的 shape 为(N,1,2)
列表b 的 shape 为 (k,2)
进行减法运算: a-b
两个列表都会扩充到(N,K,2)进行计算
这样就不用循环一个个计算了