Python计算任意多边形间的重叠面积的示例代码

时间:2022-06-05 02:54:25

简介

跟某人讨论一个排样问题。
他说,算法搜索速度很慢,每两个物体间的重叠面积计算时间若按1s来算,300个物体需要计算将近9万次。
我说,这用计算机视觉难道不是几句话解决的嘛!
(小小的嘚瑟一把,虽然做了这么久的CV,一直觉得自己一无所成,但是没想到默默的就能解决别人的问题了哈哈哈~~)

本文档目的为:
给定的数据为多边形的各个顶点,为N*2的矩阵,N 为多边形的顶点个数,计算任意两个多边形重叠面积计算的工具介绍及程序
注意,并不涉及IOU的计算(虽然只是一句话的事哈哈哈)等,只是要重叠面积。

1. shapely工具箱

判断任意两个多边形的面积是否有交集有函数A.intersects(B)实现,若A和B有交集,返回为TRUE,若没有交集,返回为FALSE
计算任意两个多边形的面积有函数A.intersection(B).area,输出直接为A和B的交集的面积。

那么问题在于,如何将多边形顶点的坐标,换为组成多边形的所有内部点的坐标。
python有个工具箱shapely。用于解决多边形有关问题。其中有个多边形填充函数Polygon.

shapely的安装方法为:

?
1
pip install Shapely

但是不知道为什么,用该方法安装时,一直报错,后来直接从https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely (直接ctrl+F搜索shapely快速转到下载位置)下载了该文件,用地址名安装。

Python计算任意多边形间的重叠面积的示例代码

安装成功后,即可用下面示意的程序计算面积。

2. 程序

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import numpy as np
import time
from shapely.geometry import Polygon  # 多边形
import scipy.io as io
 
def Cal_area_2poly(data1,data2):
    """
    任意两个图形的相交面积的计算
    :param data1: 当前物体
    :param data2: 待比较的物体
    :return: 当前物体与待比较的物体的面积交集
    """
 
    poly1 = Polygon(data1).convex_hull      # Polygon:多边形对象
    poly2 = Polygon(data2).convex_hull
 
    if not poly1.intersects(poly2):
        inter_area = 0  # 如果两四边形不相交
    else:
        inter_area = poly1.intersection(poly2).area  # 相交面积
    return inter_area
    
data1 = []  # 带比较的第一个物体的顶点坐标
data2 = []   #待比较的第二个物体的顶点坐标
area = Cal_area_2poly(data1,data2)

到此这篇关于Python计算任意多边形间的重叠面积的示例代码的文章就介绍到这了,更多相关Python 计算重叠面积内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/lovetaozibaby/article/details/102644743