debug in Insmap

opendatalab · Apr 11, 2023 · 366524a · 366524a
1 parent 7df23d1
commit 366524a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsdl/geometry/insmap.py b/dsdl/geometry/insmap.py
@@ -21,7 +21,7 @@ def visualize(self, image, palette, **kwargs):
             The image where the current instance map has been drawn on.
         """
         ins_map = self.to_array()
-        color_map = np.zeros((ins_map.shape[0], ins_map[1], 3), dtype=np.uint8)
+        color_map = np.zeros((ins_map.shape[0], ins_map.shape[1], 3), dtype=np.uint8)
         ins_ids = np.unique(ins_map)
 
         for ins_id in ins_ids: