版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/baidu_36669549/article/details/85006116
1.先求任意四个点连成四边形的面积
这个问题可以用下面的图简单的看一下
图像的坐标如上图所示,大致的想法就是四个点可以确定四条线,然后进行判断,在红色区域中则为面积中的一个像素,否则不在。先求四条线的斜率
def line_slope(x1,y1,x2,y2,x3,y3,x4,y4):
k1=(y2-y1)/(x2-x1)
k2=(y3-y2)/(x3-x2)
k3=(y4-y3)/(x4-x3)
k4=(y1-y4)/(x1-x4)
return k1,k2,k3,k4
然后计算每个位置上的各个函数值
l1=int(tk1*(i-tx1)+ty1)
l2=int(tk2*(i-tx2)+ty2)
l3=int(tk3*(i-tx3)+ty3)
l4=int(tk4*(i-tx4)+ty4)
判断条件很重要,因为左边是那样排列的,所以判断条件就是
(l1<=j)&(l2>=j)&(l3>=j)&(l4<=j)
也就是在红色区域中任取一点都满足这个条件。定义一个全局变量,满足条件就+1。面积就求出来了。
其实求面积并不是我的目的
2.求相交的面积
两个面积分别求出来以后,两个面积的交集面积最简单的可以通过对照两个区域的坐标进行求解。
也就是在分别计算两个面积的时候记下符合条件的坐标(x,y)存放到数组中,最后比较两个数组中相等的元素的个数即可求解。
3.并面积
交的面积计算完后,可以用下面的公式(S1:四边形1的面积、S2:四边形2的面积、iu:交面积)
并面积=S1-iu+S2
4.IOU
交面积/并面积
5.测试
(1)
(2)
(3)
可以看到最后一种情况红框已经把ground truth包含了,所以可以直接用S2/S1来检验算法的准确性
而13882/19307=0.719013829181126,可以证明算法的准确性。
测试代码:链接
和别的算法进行比较
还是有区别的。。。
v2是向量方法计算
v3是shapely包计算