赞
踩
dota_devkit源码
DOTA dataset
干货 | 目标检测入门,看这篇就够了(已更完)
整个devkit中,最关键的代码之一在文件./polyiou.cpp
之中。这个.cpp文件提供了一套高效的对两个任意多边形计算iou(intersection over union)的算法。这为我们在detection阶段实现精确选框提供了帮助。下文开始介绍整个.cpp文件内容及其用到的算法。
1.
const double eps=1E-8;
int sig(double d){
return(d>eps)-(d<-eps);
}
判断两个浮点数的大小关系,阈值由const double eps=1E-8;
确定。
2.
struct Point{
double x,y; Point(){}
Point(double x,double y):x(x),y(y){}
bool operator==(const Point&p)const{
return sig(x-p.x)==0&&sig(y-p.y)==0;
}
};
一个结构体,提供一个直角坐标系中点的对象,内置方法有创建
以及是否为同一点
的判定。
3.
double cross(Point o,Point a,Point b){ //叉积
return(a.x-o.x)*(b.y-o.y)-(b.x-o.x)*(a.y-o.y);
}
进行一次有向叉积操作,返回三角形OAB的有向面积。面积的正负由向量OA和OB的旋转关系确定,若由OA致OB为逆时针旋转,则为正,反之为负。叉积示意图:
4.
double area(Point* ps,int n){
ps[n]=ps[0];
double res=0;
for(int i=0;i<n;i++){
res+=ps[i].x*ps[i+1].y-ps[i].y*ps[i+1].x;
}
return res/2.0;
}
参数Point* ps
是多边形各顶点坐标,参数int n
指定多边形的边数。这里ps[n]=ps[0];
是为了下面的循环结构能方便的取到起始点的值。
累加结构内部实际上也是一种叉积运算,并将结果依次累加。由于多边形的顶点是按照某一时钟顺序输入的,并且叉积产生的是有向面积,所以以上图求四边形面积为例,最终返回的面积值是
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。