吴恩达深度学习:class4week3 作业bug

compute iou

在我看的参考作业里面关于iou种inner_area的计算是如下所示的:

    xi1 = max(box1[0], box2[0])
    yi1 = max(box1[1], box2[1])
    xi2 = min(box1[2], box2[2])
    yi2 = min(box1[3], box2[3])
    inter_area = (xi2 - xi1) * (yi2 - yi1)

但是实际运行的时候报错,意思是max函数不能这样用
虽然这是其中一个问题,但是我感觉这个样子计算还是没有任何意义
于是我自己将计算改为:

    xi = max(box1[0], box2[0])
    yi = max(box1[1], box2[1])
    w = min(box1[2], box2[2]) - xi
    h = min(box1[3], box2[3]) - yi
    if w < 0 or h < 0:
        inner_area = 0
    else:
        inter_area = h * w

这样就可以了,感觉也可以包括那种inner_area=0的情况

你可能感兴趣的:(吴恩达深度学习:class4week3 作业bug)