成功解决shapely.errors.GEOSException: TopologyException: Input geom 1 is invalid: Self-intersection……

项目场景：

求IOU时出现问题

---

问题描述

在使用shapely用如下代码：

def compute_iou(polygon1, polygon2):
     # 创建多边形对象
    poly1 = Polygon(polygon1)
    poly2 = Polygon(polygon2)
    # 计算两个多边形的交集
    intersection = poly1.intersection(poly2)
    
    # 计算交集的面积
    intersection_area = intersection.area
    
    # 计算两个多边形的面积
    area1 = poly1.area
    area2 = poly2.area
    
    # 求面积较小的多边形
    min_area_polygon = 1 if area1 < area2 else 2
    
    # 计算IoU
    iou = intersection_area / min(area1, area2)
    
    return iou, min_area_polygon

求两个多边形的iou值时，报错如下：

---

原因分析：

经查，是因为有多边形出现了自相交的情况，即多边形的某一部分和其自身相交

---

解决方案：

1、用shapely.geometry.shape(geom).is_valid函数来判断多边形是否有效，如果无效直接过滤掉，比较暴力，不推荐
2、通过buffer方法来平滑图形，该方法可以将多边形扩大或缩小，修改后函数如下：

def compute_iou(polygon1, polygon2):
     # 创建多边形对象
    poly1 = Polygon(polygon1).buffer(0.01)
    poly2 = Polygon(polygon2).buffer(0.01)
    # 计算两个多边形的交集
    intersection = poly1.intersection(poly2)
    
    # 计算交集的面积
    intersection_area = intersection.area
    
    # 计算两个多边形的面积
    area1 = poly1.area
    area2 = poly2.area
    
    # 计算面积较小的多边形
    min_area_polygon = 1 if area1 < area2 else 2
    
    # 计算IoU
    iou = intersection_area / min(area1, area2)
    
    return iou, min_area_polygon

通过如上代码，可解决我的问题，大家可以根据自己的情况去尝试。