Skip to content

Commit d371162

Browse files
committed
suppress shapely warning
1 parent 2ecf4a4 commit d371162

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

mmdet/structures/mask/structures.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -757,6 +757,10 @@ def crop(self, bbox):
757757
# reference: https://github.com/facebookresearch/fvcore/blob/main/fvcore/transforms/transform.py # noqa
758758
crop_box = geometry.box(x1, y1, x2, y2).buffer(0.0)
759759
cropped_masks = []
760+
# suppress shapely warnings util it incorporates GEOS>=3.11.2
761+
# reference: https://github.com/shapely/shapely/issues/1345
762+
initial_settings = np.seterr()
763+
np.seterr(invalid='ignore')
760764
for poly_per_obj in self.masks:
761765
cropped_poly_per_obj = []
762766
for p in poly_per_obj:
@@ -785,10 +789,11 @@ def crop(self, bbox):
785789
coords[:, 0] -= x1
786790
coords[:, 1] -= y1
787791
cropped_poly_per_obj.append(coords.reshape(-1))
788-
792+
# a dummy polygon to avoid misalignment between masks and boxes
789793
if len(cropped_poly_per_obj) == 0:
790794
cropped_poly_per_obj = [np.array([0, 0, 0, 0, 0, 0])]
791795
cropped_masks.append(cropped_poly_per_obj)
796+
np.seterr(**initial_settings)
792797
cropped_masks = PolygonMasks(cropped_masks, h, w)
793798
return cropped_masks
794799

0 commit comments

Comments
 (0)