diou ciou python计算方法

目录

numpy计算法方法：

diou比ciou要小，比iou的值也小，训练时候用比较好。

iou ciou比较代码：

jcard跟iou结果一样,跟第一个iou不一样，还需要研究。

这个是完整的计算diou ciou代码：

---

numpy计算法方法：

和下面pytorch的能对上:

IoU、GIoU、DIoU、CIoU详解及python代码实现_牛虎兔的博客-CSDN博客_python giou计算

import math


def euclidean_distance(p1, p2):
    '''
    计算两个点的欧式距离
    '''
    x1, y1 = p1
    x2, y2 = p2
    return math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)


class BBox:
    def __init__(self, x, y, r, b):
        '''
        定义框，左上角及右下角坐标
        '''
        self.x, self.y, self.r, self.b = x, y, r, b

    def __xor