【CV基石】计算两个矩形框的IOU

最新推荐文章于 2024-07-29 22:34:10 发布

iamrealAI

最新推荐文章于 2024-07-29 22:34:10 发布

阅读量2.3k

点赞数 1

分类专栏：计算机视觉文章标签：计算机视觉目标检测感知计算机器学习人工智能

本文链接：https://blog.csdn.net/aaronyking/article/details/89743743

版权

本文深入探讨了在计算机视觉领域中，如何计算两个矩形框的交并比（Intersection Over Union, IOU），它是目标检测任务中的关键指标。通过理解IOU，我们可以更好地评估检测算法的性能，并在机器学习模型训练中设置阈值。" 101023760,7704498,NumPy进阶技巧解析,"['numpy', 'python', '数据科学', '数组操作', '向量化']

摘要由CSDN通过智能技术生成

# -*- coding: UTF-8 -*-
import numpy as np
import cv2

def iou(bbox1, bbox2, center=False):
    """Compute the iou of two boxes.
    Parameters
    ----------
    bbox1, bbox2: list.
        The bounding box coordinates: [xmin, ymin, xmax, ymax] or [xcenter, ycenter, w, h].
    center: str, default is 'False'.
        The format of coordinate.
        center=False: [xmin, ymin, xmax, ymax]
        center=True: [xcenter, ycenter, w, h]
    Returns
    -------
    iou: float.
        The iou of bbox1 and bbox2.
    """
    if center == False:
        xmin1, ymin1, xmax1, ymax1 = bbox1
        xmin2, ymin2, xmax2, ymax2 = bbox2
    else:
        xmin1, ymin1 = int(bbox1[0] - bbox1[2] / 2.0), int(bbox1[1] - bbox1[3] / 2.0)
        xmax1, ymax1 = int(bbox1[0] + bbox1[