python shapely.geometry.polygon任意两个四边形的IOU计算实例
作者:不吃鱼的猫~ 发布时间:2021-09-23 03:19:46
标签:python,四边形,IOU
在目标检测中一个很重要的问题就是NMS及IOU计算,而一般所说的目标检测检测的box是规则矩形框,计算IOU也非常简单,有两种方法:
1. 两个矩形的宽之和减去组合后的矩形的宽就是重叠矩形的宽,同比重叠矩形的高
2. 右下角的minx减去左上角的maxx就是重叠矩形的宽,同比高
然后 IOU = 重叠面积 / (两矩形面积和—重叠面积)
然,不规则四边形就不能通过这种方式来计算,找了好久数学资料,还是没找到答案(鄙人数学渣渣),最后看了白翔老师的textBoxes++论文源码后,知道python的shapely包可以直接做到,下面给出的代码和注释:
import numpy as np
import shapely
from shapely.geometry import Polygon,MultiPoint #多边形
line1=[2,0,2,2,0,0,0,2] #四边形四个点坐标的一维数组表示,[x,y,x,y....]
a=np.array(line1).reshape(4, 2) #四边形二维坐标表示
poly1 = Polygon(a).convex_hull #python四边形对象,会自动计算四个点,最后四个点顺序为:左上 左下 右下 右上 左上
print(Polygon(a).convex_hull) #可以打印看看是不是这样子
line2=[1,1,4,1,4,4,1,4]
b=np.array(line2).reshape(4, 2)
poly2 = Polygon(b).convex_hull
print(Polygon(b).convex_hull)
union_poly = np.concatenate((a,b)) #合并两个box坐标,变为8*2
#print(union_poly)
print(MultiPoint(union_poly).convex_hull) #包含两四边形最小的多边形点
if not poly1.intersects(poly2): #如果两四边形不相交
iou = 0
else:
try:
inter_area = poly1.intersection(poly2).area #相交面积
print(inter_area)
#union_area = poly1.area + poly2.area - inter_area
union_area = MultiPoint(union_poly).convex_hull.area
print(union_area)
if union_area == 0:
iou= 0
#iou = float(inter_area) / (union_area-inter_area) #错了
iou=float(inter_area) / union_area
# iou=float(inter_area) /(poly1.area+poly2.area-inter_area)
# 源码中给出了两种IOU计算方式,第一种计算的是: 交集部分/包含两个四边形最小多边形的面积
# 第二种: 交集 / 并集(常见矩形框IOU计算方式)
except shapely.geos.TopologicalError:
print('shapely.geos.TopologicalError occured, iou set to 0')
iou = 0
print(a)
print(iou)
具体原理还没弄明白,还在研究中,研究完再给出来(当然数学渣渣能不能研究出来有待商榷*—*)
补充知识:python 二维坐标多边形 计算多边形中心点,以及距该中心点最远的距离
我就废话不多说了,还是直接看代码吧!
def center_geolocation(geolocations):
'''
输入多个经纬度坐标(格式:[[lon1, lat1],[lon2, lat2],....[lonn, latn]]),找出中心点
:param geolocations:
:return:中心点坐标 [lon,lat]
'''
#求平均数 同时角度弧度转化 得到中心点
x = 0# lon
y = 0# lat
z = 0
lenth = len(geolocations)
for lon, lat in geolocations:
lon = radians(float(lon))
# radians(float(lon)) Convert angle x from degrees to radians
# 把角度 x 从度数转化为 弧度
lat = radians(float(lat))
x += cos(lat) * cos(lon)
y += cos(lat) * sin(lon)
z += sin(lat)
x = float(x / lenth)
y = float(y / lenth)
z = float(z / lenth)
return (degrees(atan2(y, x)), degrees(atan2(z, sqrt(x * x + y * y))))
#得到离中心点里程最近的里程
def geodistance(lon1,lat1,lon2,lat2):
'''
得到两个经纬度坐标距离 单位为千米 (计算不分前后顺序)
:param lon1: 第一个坐标 维度
:param lat1: 第一个坐标 经度
:param lon2: 第二个坐标 维度
:param lat2: 第二个坐标 经度
:return: distance 单位千米
'''
# lon1,lat1,lon2,lat2 = (120.12802999999997,30.28708,115.86572000000001,28.7427)
lon1, lat1, lon2, lat2 = map(radians, [float(lon1), float(lat1), float(lon2), float(lat2)]) #经纬度转换成弧度
dlon=lon2-lon1
dlat=lat2-lat1
a=sin(dlat/2)**2 + cos(lat1) * cos(lat2) * sin(dlon/2)**2
distance=2*asin(sqrt(a))*6371*1000 #地球平均半径,6371km
distance=round(distance/1000,3)
print(distance)
return distance
def getMaxestDistance(geolocations,centre):
'''
中心点 距离 多个经纬度左边 最远的距离
:param geolocations: 多个经纬度坐标(格式:[[lon1, lat1],[lon2, lat2],....[lonn, latn]])
:param centre: 中心点 centre [lon,lat]
:return: 最远距离 千米
'''
distantces=[]
for lon, lat in geolocations:
d=geodistance(lat,lon,centre[1],centre[0])
distantces.append(d)
# print(distantces)
return max(distantces)
def getOnePolyygen(geolocations):
'''
输入多个经纬度坐标(格式:[[lon1, lat1],[lon2, lat2],....[lonn, latn]]),找出距该多边形中心点最远的距离
:param geolocations:多个经纬度坐标(格式:[[lon1, lat1],[lon2, lat2],....[lonn, latn]])
:return:center,neartDistance 多边形中心点 最远距离
'''
center=center_geolocation(geolocations) # 得到中心点
neartDistance=getMaxestDistance(geolocations,center)
# print(center,"-----------------",neartDistance)
return center,neartDistance
来源:https://blog.csdn.net/u012433049/article/details/82909484
0
投稿
猜你喜欢
- 1、算术运算符:+、-、*、/、%。2、递增/递减运算符:如$a++,$a--,++$a,--$a.如:<?php$a=10;$b=5
- 参数Parameters解析响应时间resolveTimeout 数据类型:长整型。简单地说就是程序对目标主机的名字解析解析的一个过程时间。
- Sometimes we have need to interact with an application,for examp
- 本节列出了可以运行MySQL的操作系统。我们使用GNU Autoconf,因此将MySQL移植到所有使用Posix线程和C++编译器的现代系
- Python 风格规范(Google)本项目并非 Google 官方项目, 而是由国内程序员凭热情创建和维护。如果你关注的是 Google
- jQuery之所以如此流行并被从大公司到个人博客的几乎每个人都广泛使用,是因为它上手和使用相当简单,而且为我们提供了一些人都不知道的相当棒的
- 环境:Ubuntu14.04,tensorflow=1.4(bazel源码安装),Anaconda python=3.6声明变量主要有两种方
- 最近项目使用c++操作Python脚本,选用boost.python库。在window下编译安装很顺利,但是在Linux下一直编译不通过,总
- 在使用Django做前端后端项目时,登陆认证方法往往使用的是jwt_token,但是想自定义登陆成功和失败的返回体。1.当用户名和密码正确就
- 图片轮播滤镜转换效果,只支持ie浏览器<img width="156" height="60"
- 本文实例讲述了Python基于SMTP协议实现发送邮件功能。分享给大家供大家参考,具体如下:SMTP(Simple Mail Transfe
- 本文为大家分享了Eclipse开发python脚本的具体方法,供大家参考,具体内容如下一、安装python1.访问网址,可以看到如下图所示界
- set oSQLServer =server.createobject("SQLDMO.SQLServer"
- 对于PyQT4, PyQT5在一些使用上有着比较明显的变化有很大的变化,让人惊讶是在emit和connect上的一些变化比较有意思,相信也是
- 两组字符串数据,需要比较其中相同的数据,并将其值相加并组成一个新的字符串数据a1="sp2=20;sp1=34;"a2=
- 1,模块命名(1)模块推荐使用小写命名,(2)除非有很多字母,尽量不要用下划线因为很多模块文件存与模块名称一致的类,模块采用小写,类采用首字
- 最近由于单位数据库硬盘空间不足,整理的时候查了许多文章,也进行了测试,整理后得出一些经验供大家参考。首先,在网上看到一篇文章,如何Shrin
- 大家都遇到过验证码,随着灌水机的发展,验证码也是日新月异,验证码开始分了繁体简体,带着字母、符号,甚至开始了中文验证码,看到这样的验证码,估
- 如何做一个专门显示文本文件的页面? 代码如下:txt.asp<html><head&g
- 最近在做图片相关的工作,面对各种格式的图片转换,在OpenCV跟Pillow之间来回切换。觉得有必要把这些东西记录下来,以备不时之需。首先介