lir

lir

快速计算二进制网格最大内部矩形的Python库

lir是一个开源Python库,用于快速计算二进制网格中的最大内部矩形。该项目利用Numba编译技术提高执行速度,支持基于网格和轮廓的计算方法。lir不仅适用于二进制网格,还能处理多边形坐标列表,为图像处理和计算几何领域提供了高效解决方案。项目文档包含详细使用说明和性能优化建议,方便开发者快速上手。

Largest Interior Rectangle二进制网格PythonNumba算法优化Github开源项目

lir

在二进制网格中快速计算最大内部矩形。

示例1

:rocket: 通过Numba,Python代码被编译成机器码,以原生机器码速度执行!

安装

使用pip从PyPI安装largestinteriorrectangle。

pip install largestinteriorrectangle

:snail: 编译代码需要一些时间(在我的电脑上大约1分钟)。这只需在安装后执行一次,编译后的代码会被缓存。如果你想在应用程序外进行此操作,打开Python控制台并输入import largestinteriorrectangle。你会看到控制台暂时阻塞。完成后,第二次导入只需几毫秒。

使用方法

import largestinteriorrectangle as lir import numpy as np grid = np.array([[0, 0, 1, 0, 0, 0, 0, 0, 0], [0, 0, 1, 0, 1, 1, 0, 0, 0], [0, 0, 1, 1, 1, 1, 1, 0, 0], [0, 0, 1, 1, 1, 1, 1, 1, 0], [0, 0, 1, 1, 1, 1, 1, 1, 0], [0, 1, 1, 1, 1, 1, 1, 0, 0], [0, 0, 1, 1, 1, 1, 0, 0, 0], [0, 0, 1, 1, 1, 1, 0, 0, 0], [1, 1, 1, 1, 1, 1, 0, 0, 0], [1, 1, 0, 0, 0, 1, 1, 1, 1], [0, 0, 0, 0, 0, 0, 0, 0, 0]], "bool") lir.lir(grid) # array([2, 2, 4, 7])

对于较大的网格,指定要考虑的多边形轮廓可以显著提高性能。如果网格只有一个多边形,如示例所示,可以这样获取轮廓(使用opencv):

import cv2 as cv cv_grid = grid.astype("uint8") * 255 contours, _ = \ cv.findContours(cv_grid, cv.RETR_TREE, cv.CHAIN_APPROX_NONE) contour = contours[0][:, 0, :]

然后计算矩形:

lir.lir(grid, contour) # array([2, 2, 4, 7])

你也可以从多边形坐标列表计算lir:

import numpy as np import cv2 as cv import largestinteriorrectangle as lir polygon = np.array([[[20, 15], [210, 10], [220, 100], [100, 150], [20, 100]]], np.int32) rectangle = lir.lir(polygon) img = np.zeros((160, 240, 3), dtype="uint8") cv.polylines(img, [polygon], True, (0, 0, 255), 1) cv.rectangle(img, lir.pt1(rectangle), lir.pt2(rectangle), (255, 0, 0), 1) cv.imshow('lir', img) cv.waitKey(0) cv.destroyAllWindows()

从多边形

在后台,使用cv.fillPoly创建网格(需要OpenCV作为可选依赖),然后计算轮廓并使用基于轮廓的lir。

另请参阅我在这个Stack Overflow问题中的回答

贡献

欢迎提交拉取请求。对于重大更改,请先开issue讨论你想要改变的内容。

请确保适当更新测试。

使用以下命令运行测试:

python -m unittest

许可证

Apache License 2.0

致谢

感谢Tim Swan将他的C#版最大内部矩形实现开源,并写了一篇很棒的博客文章。第一部分主要是用Python重新实现他的解决方案。

使用的算法在2019年的论文Algorithm for finding the largest inscribed rectangle in polygon中由Zahraa Marzeh, Maryam Tahmasbi和Narges Mireh描述。

还要感谢Mark Setchelljoni,他们在这个Stack Overflow问题中极大地帮助优化了使用cpython/numba的性能。

工作原理

对于二进制网格:

网格

我们可以为每个单元格指定向右和向下可以走多远:

水平邻接垂直邻接
水平邻接垂直邻接

现在目标是为每个单元格找到可能的矩形。为此,我们可以基于水平邻接指定水平向量,基于垂直邻接指定垂直向量:

水平向量 (2,2)垂直向量 (2,2)
水平向量垂直向量

所以在给定的单元格(2,2)处,水平向量是(5,4),垂直向量是(7,4)。

反转任一向量可以通过堆叠向量创建跨度,例如,将垂直向量反转为(4,7)会得到一组(5 x 4)和(4 x 7)的跨度。 由于47=28 > 54=20,宽度为4、高度为7的矩形是单元格(2,2)可能的最大矩形。宽度和高度存储在跨度图中,其中所有单元格的最大矩形的宽度和高度都被记录。通过面积,我们可以确定(2,2)处最大的矩形,其宽度为4,高度为7。

宽度高度面积
跨度图宽度跨度图高度跨度图面积

基于轮廓的LIR

特别是对于更大的网格,通过仅分析多边形的轮廓可以进一步优化功能。以下是通过计算不同分辨率掩码的lir所得到的时间:

时间时间(对数转换)
性能比较性能比较对数

通过仅分析轮廓像素而不是所有单元格,节省了计算成本。我们利用了LIR总是接触多边形轮廓的事实。以下是其工作原理:

网格

轮廓单元格可以向一个(蓝色)、两个(绿色)或三个(红色)方向(上、下、左、右)延伸:

方向图

通过计算所有可能方向的跨度,我们可以得到一个跨度图:

宽度高度面积
跨度图宽度跨度图高度跨度图面积

为了分析这里发生的情况,我们将仔细查看单元格(4,2)。

单元格2_2方向图

它可以向3个方向延伸:左、下和右。向左和下延伸,最大跨度为(3 x 7)。最终跨度以从左到右和从上到下的表示法记录。然而,在这种情况下,宽度是从右到左计算的。我们可以用简单的公式x = 单元格_x - 跨度宽度 + 1转换它,在这种情况下是4 - 3 + 1 = 2。由于高度已经从上到下计算,y不变,跨度(3 x 7)被分配到单元格(2,2)(黑色虚线)。

(2,2)是跨度图中面积最大的单元格(除了(1,6))。然而,矩形可以向右扩展(青色虚线)的信息丢失了。

因此,对于像(2,2)这样不在轮廓上且来自可向3个方向延伸的轮廓单元格的"候选单元格",我们创建一个新的跨度图(使用从左到右和从上到下的邻接关系):

候选图

宽度高度面积
跨度图2宽度跨度图2高度跨度图2面积

比较两个跨度图的最大跨度,返回较大的一个作为lir,在这种情况下是单元格(2,2),跨度为(4 x 7)。

编辑推荐精选

iTerms

iTerms

企业专属的AI法律顾问

iTerms是法大大集团旗下法律子品牌,基于最先进的大语言模型(LLM)、专业的法律知识库和强大的智能体架构,帮助企业扫清合规障碍,筑牢风控防线,成为您企业专属的AI法律顾问。

SimilarWeb流量提升

SimilarWeb流量提升

稳定高效的流量提升解决方案,助力品牌曝光

稳定高效的流量提升解决方案,助力品牌曝光

Sora2视频免费生成

Sora2视频免费生成

最新版Sora2模型免费使用,一键生成无水印视频

最新版Sora2模型免费使用,一键生成无水印视频

Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

AI助手热门AI工具AI创作AI辅助写作讯飞绘文内容运营个性化文章多平台分发
TRAE编程

TRAE编程

AI辅助编程,代码自动修复

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

热门AI工具生产力协作转型TraeAI IDE
商汤小浣熊

商汤小浣熊

最强AI数据分析助手

小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。

imini AI

imini AI

像人一样思考的AI智能体

imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。

Keevx

Keevx

AI数字人视频创作平台

Keevx 一款开箱即用的AI数字人视频创作平台,广泛适用于电商广告、企业培训与社媒宣传,让全球企业与个人创作者无需拍摄剪辑,就能快速生成多语言、高质量的专业视频。

即梦AI

即梦AI

一站式AI创作平台

提供 AI 驱动的图片、视频生成及数字人等功能,助力创意创作

下拉加载更多