debug InsMap

opendatalab · Apr 11, 2023 · 198bcc8 · 198bcc8
1 parent fb95c49
commit 198bcc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsdl/geometry/insmap.py b/dsdl/geometry/insmap.py
@@ -27,7 +27,7 @@ def visualize(self, image, palette, **kwargs):
         for ins_id in ins_ids:
             if ins_id == 0:
                 continue
-            contour_color = tuple(np.random.randint(0, 255, size=[3]))
+            contour_color = tuple(int(_) for _ in np.random.randint(0, 255, size=[3]))
             this_map = (ins_map == ins_id).astype(np.uint8) * 255
             contours, _ = cv2.findContours(this_map, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
             cv2.drawContours(color_map, contours, -1, contour_color, 2)