DOTA_devkit 原理浅析

dota_devkit源码
DOTA dataset
干货 | 目标检测入门，看这篇就够了（已更完）

---

./polyiou.cpp

整个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];是为了下面的循环结构能方便的取到起始点的值。

累加结构内部实际上也是一种叉积运算，并将结果依次累加。由于多边形的顶点是按照某一时钟顺序输入的，并且叉积产生的是有向面积，所以以上图求四边形面积为例，最终返回的面积值是$area=\triangle OAB+\triangle OBC+\triangle OCD+\triangle ODC$。其中 △OAB<