iou-giou-diou-ciou-nms相关code

import math

import numpy


def iou(box1,box2):
     x1,y1,x2,y2=box1
     x3,y3,x4,y4=box2
     area1=max(0,x2-x1+1)*max(0,y2-y1+1)
     area2=max(0,x4-x3+1)*max(0,y4-y3+1)

     xx1=max(x1,x3)
     yy1=max(y1,y3)
     xx2=min(x2,x4)
     yy2=min(y2,y4)

     w=max(0,xx2-xx1+1)
     h=max(0,yy2-yy1+1)
     inter=w*h
     u=area1+area2-inter

     return inter/u

def giou(box1,box2):
    x1, y1, x2, y2 = box1
    x3, y3, x4, y4 = box2
    area1 = max(0, x2 - x1 + 1) * max(0, y2 - y1 + 1)
    area2 = max(0, x4 - x3 + 1) * max(0, y4 - y3 + 1)

    xx1 = max(x1, x3)
    yy1 = max(y1, y3)
    xx2 = min(x2, x4)
    yy2 = min(y2, y4)
    ae_x1=min(x1,x3)
    ae_y1=min(y1,y3)
    ae_x2=max(x2,x4)
    ae_y2=max(y2,y4)

    inter=max(0,xx2-xx1+1)*max(0,yy2-yy1+1)
    u = area1 + area2 - inter
    ae=max(0,ae_x2-ae_x1+1)*max(ae_y2-ae_y1+1)

    return inter/u-(ae-u)/ae

def diou(box1,box2):
    x1, y1, x2, y2 = box1
    x3, y3, x4, y4 = box2
    area1 = max(0, x2 - x1 + 1) * max(0, y2 - y1 + 1)
    area2 = max(0, x4 - x3 + 1) * max(0, y4 - y3 + 1)

    c1_x,c1_y=(x2+x1)/2,(y2+y1)/2
    c2_x,c2_y=(x4+x3)/2,(y4+y3)/2
    p2=(c2_x-c1_x)*(c2_x-c1_x)+(c2_y-c1_y)*(c2_y-c1_y)

    xx1 = max(x1, x3)
    yy1 = max(y1, y3)
    xx2 = min(x2, x4)
    yy2 = min(y2, y4)
    ae_x1=min(x1,x3)
    ae_y1=min(y1,y3)
    ae_x2=max(x2,x4)
    ae_y2=max(y2,y4)
    inter=max(0,xx2-xx1+1)*max(0,yy2-yy1+1)
    u = area1 + area2 - inter
    c2=(ae_x2-ae_x1)*(ae_x2-ae_x1)+(ae_y2-ae_y1)*(ae_y2-ae_y1)

    return inter/u-p2/c2


def ciou(box1, box2,alpha):
    x1, y1, x2, y2 = box1
    x3, y3, x4, y4 = box2
    w1,h1=max(x2-x1+1,0),max(y2-y1+1,1e-15)
    w2, h2 = max(x4 - x3 + 1, 0), max(y4 - y3 + 1, 1e-15)
    area1 = max(0, x2 - x1 + 1) * max(0, y2 - y1 + 1)
    area2 = max(0, x4 - x3 + 1) * max(0, y4 - y3 + 1)

    c1_x, c1_y = (x2 + x1) / 2, (y2 + y1) / 2
    c2_x, c2_y = (x4 + x3) / 2, (y4 + y3) / 2
    p2 = (c2_x - c1_x) * (c2_x - c1_x) + (c2_y - c1_y) * (c2_y - c1_y)

    xx1 = max(x1, x3)
    yy1 = max(y1, y3)
    xx2 = min(x2, x4)
    yy2 = min(y2, y4)
    ae_x1 = min(x1, x3)
    ae_y1 = min(y1, y3)
    ae_x2 = max(x2, x4)
    ae_y2 = max(y2, y4)
    inter = max(0, xx2 - xx1 + 1) * max(0, yy2 - yy1 + 1)
    u = area1 + area2 - inter
    c2 = (ae_x2 - ae_x1) * (ae_x2 - ae_x1) + (ae_y2 - ae_y1) * (ae_y2 - ae_y1)

    v=4*((numpy.arctan(w1/h1)-numpy.arctan(w2/h2))**2)/(numpy.pi**2)

    return inter / u - p2 / c2-alpha*v

def nms(dets,thre):
    x1,y1,x2,y2,s=dets[:0],dets[:1],dets[:2],dets[:3],dets[:4]
    order=numpy.argsort(s)[::-1]
    keep=[]
    area=(x2-x1+1)*(y2-y1+1)

    while order:
        i=order[0]
        keep.append(i)

        xx1=numpy.maximum(x1[i],x1[order[1:]])
        yy1 = numpy.maximum(y1[i], y1[order[1:]])
        xx2 = numpy.minimum(x2[i], x2[order[1:]])
        yy2 = numpy.minimum(y2[i], y2[order[1:]])

        w=numpy.maximum(xx2-xx1+1,0)
        h=numpy.maximum(yy2-yy1+1,0)
        inter=w*h
        iou=inter/(area[i]+area[order[1:]]-inter)
        inds=numpy.where(iou<=thre)[0]
        order=order[inds+1]
    return  keep