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)。

编辑推荐精选

TRAE编程

TRAE编程

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

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

AI工具TraeAI IDE协作生产力转型热门
蛙蛙写作

蛙蛙写作

AI小说写作助手,一站式润色、改写、扩写

蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。

AI辅助写作AI工具蛙蛙写作AI写作工具学术助手办公助手营销助手AI助手
问小白

问小白

全能AI智能助手,随时解答生活与工作的多样问题

问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。

热门AI助手AI对话AI工具聊天机器人
Transly

Transly

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

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

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
讯飞星火

讯飞星火

深度推理能力全新升级,全面对标OpenAI o1

科大讯飞的星火大模型,支持语言理解、知识问答和文本创作等多功能,适用于多种文件和业务场景,提升办公和日常生活的效率。讯飞星火是一个提供丰富智能服务的平台,涵盖科技资讯、图像创作、写作辅助、编程解答、科研文献解读等功能,能为不同需求的用户提供便捷高效的帮助,助力用户轻松获取信息、解决问题,满足多样化使用场景。

热门AI开发模型训练AI工具讯飞星火大模型智能问答内容创作多语种支持智慧生活
Spark-TTS

Spark-TTS

一种基于大语言模型的高效单流解耦语音令牌文本到语音合成模型

Spark-TTS 是一个基于 PyTorch 的开源文本到语音合成项目,由多个知名机构联合参与。该项目提供了高效的 LLM(大语言模型)驱动的语音合成方案,支持语音克隆和语音创建功能,可通过命令行界面(CLI)和 Web UI 两种方式使用。用户可以根据需求调整语音的性别、音高、速度等参数,生成高质量的语音。该项目适用于多种场景,如有声读物制作、智能语音助手开发等。

咔片PPT

咔片PPT

AI助力,做PPT更简单!

咔片是一款轻量化在线演示设计工具,借助 AI 技术,实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT,提供海量模板、智能美化、素材替换等功能,适用于销售、教师、学生等各类人群,能高效制作出高品质 PPT,满足不同场景演示需求。

讯飞绘文

讯飞绘文

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

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

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

材料星

专业的AI公文写作平台,公文写作神器

AI 材料星,专业的 AI 公文写作辅助平台,为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能,支持 30 + 文稿类型生成,助力快速完成领导讲话、工作总结、述职报告等材料,提升办公效率,是体制打工人的得力写作神器。

下拉加载更多