IoU计算

输入:

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)
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值