From bd8ace712edc1130fe4677698318da1a7a8aa339 Mon Sep 17 00:00:00 2001 From: Josh Veitch-Michaelis Date: Wed, 12 Oct 2022 22:52:32 +0200 Subject: [PATCH] fix shapely deprecation warning --- fvcore/transforms/transform.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/fvcore/transforms/transform.py b/fvcore/transforms/transform.py index fbe2a58..7c18f67 100644 --- a/fvcore/transforms/transform.py +++ b/fvcore/transforms/transform.py @@ -721,6 +721,10 @@ def apply_polygons(self, polygons: list) -> list: if not isinstance(cropped, geometry.collection.BaseMultipartGeometry): cropped = [cropped] # one polygon may be cropped to multiple ones + if isinstance(cropped, geometry.MultiPolygon) or isinstance( + cropped, geometry.GeometryCollection + ): + cropped = cropped.geoms for poly in cropped: # It could produce lower dimensional objects like lines or # points, which we want to ignore