IoU计算

输入:vector<float>p1;//y1min,x1min,y1max,x1max;vector<float>p2;//y2min,x2min,y2max,x2max;输出:floatIoU代码:floatcomputeIoU(vector<float>p1,vector<float>p2){…

大家好,又见面了,我是你们的朋友全栈君。

输入:

vector<float> p1;  //y1min, x1min, y1max, x1max;
vector<float> p2;  //y2min, x2min, y2max, x2max;

输出:

float IoU

代码:

float computeIoU(vector<float> p1, vector<float> p2)
{ 
   
	//如果存在一个轴上,某个框的最小坐标大于另外一个框的最大坐标,则两框无重合。
	if(p2[3]-p1[1]<0)||(p1[3]-p2[1]<0)||(p1[2]-p2[0]<0)||(p2[2]-p1[0]<0)
		return 0;
	//两框必有重合,计算重合面积
	float h,w;
	if(p1[0]<p2[0]){ 
   
		if(p1[2]>p2[2]) h=p2[2]-p2[0];
		else h=p1[2]-p2[0];
		}
	else{ 
   
		if(p1[2]>p2[2]) h=p2[2]-p1[0];
		else h=p1[2]-p1[0];
		}
	if(p1[1]<p2[1]){ 
   
		if(p1[3]>p2[3]) w=p2[3]-p2[1];
		else w=p1[3]-p2[1];
		}
	else{ 
   
		if(p1[3]>p2[3]) w=p2[3]-p1[1];
		else w=p1[3]-p1[1];
		}
	// 容斥原理计算IoU
	return (h*w)/(p1[2]-p1[0])*(p1[3]-p1[1])+(p2[2]-p2[0])*(p2[3]-p2[1])-h*w)
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/128747.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号