Python

您所在的位置:网站首页 python判断两个列表有无交集 Python

Python

2023-10-19 20:45| 来源: 网络整理| 查看: 265

简介

更新: 评论里有说到凹多边形不适用,只能适用凸多边形,因此我把标题改为了凸多边形间重叠面积计算。

— — — — — — — — — — 分界线 — — — — — — — — — — — —

跟某人讨论一个排样问题。 他说,算法搜索速度很慢,每两个物体间的重叠面积计算时间若按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的安装方法为:

pip install Shapely

但是不知道为什么,用该方法安装时,一直报错,后来直接从https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely (直接ctrl+F搜索shapely快速转到下载位置)下载了该文件,用地址名安装。 在这里插入图片描述 安装成功后,即可用下面示意的程序计算面积。

2. 程序 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)


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3