From 4a7e88ea2e14b5d091b0a6f2d9d9a1da82842462 Mon Sep 17 00:00:00 2001 From: Dylan Auty Date: Thu, 29 Sep 2022 12:48:58 +0100 Subject: [PATCH] Updated segmentation dataloader to work with >255 instances (see https://github.com/ultralytics/yolov5/issues/9461 and PR #9493 of the yolov5 repo) --- seg/utils/segment/dataloaders.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/seg/utils/segment/dataloaders.py b/seg/utils/segment/dataloaders.py index f6fe642d07..7bdb668ce8 100644 --- a/seg/utils/segment/dataloaders.py +++ b/seg/utils/segment/dataloaders.py @@ -308,7 +308,8 @@ def polygons2masks(img_size, polygons, color, downsample_ratio=1): def polygons2masks_overlap(img_size, segments, downsample_ratio=1): """Return a (640, 640) overlap mask.""" - masks = np.zeros((img_size[0] // downsample_ratio, img_size[1] // downsample_ratio), dtype=np.uint8) + masks = np.zeros((img_size[0] // downsample_ratio, img_size[1] // downsample_ratio), + dtype=np.int32 if len(segments) > 255 else np.uint8) areas = [] ms = [] for si in range(len(segments)):