一.numpy.maximum
作用:
按元素比较两个列表,返回包含最大值的列表。
参数:x1,x2 是两个待比较的列表。如果x1.shape!=x2.shape 则进行广播操作然后再比较。
例子:
np.maximum([2, 3, 4], [1, 5, 2])
array([2, 5, 4])
np.maximum(np.eye(2), [0.5, 2]) # broadcasting
array([[ 1. , 2. ],
[ 0.5, 2. ]])
IOU计算中的列子:
boxes = np.array([[3,5],[2,7],[4,8]]).reshape(3,1,2)
anchors = np.array([[2,6],[1,8],[2,9],[3,10],[4,11],[4,9],[1,11],[2,19],[7,10]]).reshape(1,9,2)
#print(box1)
#print(box2)
class Iou_Same():
def __init__(self,anchors,boxes):
self.anchor_max = anchors//2
self.anchor_min = -self.anchor_max
self.box_max = boxes//2
self.box_min = -self.box_max
self.anchor_areas = anchors[...,0]*anchors[...,1]
self.box_areas = boxes[...,0]*boxes[...,1]