From 4073fa0b87ff19e1fc585832791e5c4c3a81514b Mon Sep 17 00:00:00 2001 From: Rajesh Timilsina <33575173+rajeshtims@users.noreply.github.com> Date: Fri, 4 Oct 2024 19:01:48 -0400 Subject: [PATCH] Fix UINT8 overflow for >255 segmentation masks (#16690) Co-authored-by: Glenn Jocher --- ultralytics/data/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/data/utils.py b/ultralytics/data/utils.py index b2d47b3b..eaf06533 100644 --- a/ultralytics/data/utils.py +++ b/ultralytics/data/utils.py @@ -216,7 +216,7 @@ def polygons2masks_overlap(imgsz, segments, downsample_ratio=1): ms = [] for si in range(len(segments)): mask = polygon2mask(imgsz, [segments[si].reshape(-1)], downsample_ratio=downsample_ratio, color=1) - ms.append(mask) + ms.append(mask.astype(masks.dtype)) areas.append(mask.sum()) areas = np.asarray(areas) index = np.argsort(-areas)