讲师中心 微信公众号

工具推荐

视频效率加速
剧云
剧云

中文短剧剧本制作平台

Seko
Seko

AI短剧一键生成成片

AniShort
AniShort

AI短剧协同制作平台

立刻MV
立刻MV

低成本歌曲 MV、视频生成

怎样在Python中使用NumPy计算几何图形包围盒交并比

秋婷酱_6349

秋婷酱_6349

发布时间:2026-11-10 10:42:20

|

236人浏览过

|

来源于php中文网

原创

包围盒交并比(IoU)是两个矩形框重叠面积与并集面积的比值,用于目标检测评估;其NumPy实现需用向量化计算求交集坐标、判空、算面积,输入为(N,4)的[x1,y1,x2,y2]格式ndarray。

怎样在python中使用numpy计算几何图形包围盒交并比

什么是包围盒交并比(IoU)?

IoU 是两个矩形框重叠区域面积与它们并集区域面积的比值,常用于目标检测评估。它不关心坐标系原点在哪,只依赖四个顶点或 [x_min, y_min, x_max, y_max] 形式的边界框表示。

用NumPy实现IoU计算的最简方式

核心是向量化计算:避免 for 循环,用广播机制一次性处理多个框对。关键步骤是求交集坐标、判空、算面积。

  • 输入必须是 shape = (N, 4) 的 ndarray,每行是 [x1, y1, x2, y2],且满足 x1 和 <code>y1
  • 交集左上角取较大值:np.maximum(boxes1[:, :2], boxes2[:, :2])
  • 交集右下角取较小值:np.minimum(boxes1[:, 2:], boxes2[:, 2:])
  • 交集宽高为负时说明无重叠,需用 np.clip(..., 0, None) 截断
def iou(boxes1, boxes2):
    # boxes1: (N, 4), boxes2: (M, 4)
    lt = np.maximum(boxes1[:, None, :2], boxes2[None, :, :2])  # (N, M, 2)
    rb = np.minimum(boxes1[:, None, 2:], boxes2[None, :, 2:])  # (N, M, 2)
    wh = np.clip(rb - lt, 0, None)  # (N, M, 2)
    inter = wh[:, :, 0] * wh[:, :, 1]  # (N, M)
<pre class='brush:python;toolbar:false;'>area1 = (boxes1[:, 2] - boxes1[:, 0]) * (boxes1[:, 3] - boxes1[:, 1])  # (N,)
area2 = (boxes2[:, 2] - boxes2[:, 0]) * (boxes2[:, 3] - boxes2[:, 1])  # (M,)
union = area1[:, None] + area2[None, :] - inter

return np.divide(inter, union, out=np.zeros_like(inter, dtype=float), where=union!=0)

常见错误和兼容性注意点

实际用的时候,90% 的问题出在数据格式和坐标顺序上,不是算法本身。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • boxes1 和 boxes2 若混用 [x_center, y_center, w, h] 格式,必须先转成 [x1, y1, x2, y2],否则 np.maximum 会错位
  • OpenCV 默认返回 (x, y, w, h),YOLO 输出常为归一化中心坐标 —— 这些都不能直接喂给上面的 iou 函数
  • 当某对框完全不重叠时,union 可能为 0,除零会得 nan;用 np.divide(..., where=...) 或提前掩码更安全
  • 若只需单对框 IoU(比如 boxes1.shape == (4,)),别硬套广播版,改用标量逻辑更清晰:
def iou_single(box1, box2):
    x1 = max(box1[0], box2[0])
    y1 = max(box1[1], box2[1])
    x2 = min(box1[2], box2[2])
    y2 = min(box1[3], box2[3])
    if x1 >= x2 or y1 >= y2:
        return 0.0
    inter = (x2 - x1) * (y2 - y1)
    area1 = (box1[2] - box1[0]) * (box1[3] - box1[1])
    area2 = (box2[2] - box2[0]) * (box2[3] - box2[1])
    return inter / (area1 + area2 - inter)

为什么不用 scikit-image 或 torchvision?

如果你只是做后处理或写 demo,直接用 NumPy 更轻量、无依赖、可控性强。但要注意:

立即学习“Python免费学习笔记(深入)”;

  • skimage.metrics.bounding_box_iou 输入是 (r0, c0, r1, c1)(即 y1, x1, y2, x2),Y 轴在前,容易和习惯的 x1,y1,x2,y2 搞反
  • torchvision.ops.box_iou 要求输入是 torch.Tensor,且默认在 GPU 上跑 —— 纯 NumPy 场景下引入 PyTorch 不划算
  • 自己写的 NumPy 版本可以无缝嵌入 numba.jit 加速,而第三方封装往往不支持

坐标顺序、数据类型、空框处理,这几个地方不盯紧,IoU 值就可能全错,而且错得不报错。

热门AI工具

更多
小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

讯飞智文

一款面向学习和办公场景的AI文档创作工具,可辅助生成PPT与Word文档,提高资料整理和内容制作效率。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1891

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

5264

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1849

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

31817

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3767

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3787

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1263

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

40

2026.11.06

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn