From 1bb75b337a81ce60db1a8bff1dcc817f2f87d8c1 Mon Sep 17 00:00:00 2001 From: pythonzz0622 Date: Mon, 19 Feb 2024 09:01:57 +0000 Subject: [PATCH] [Fix] Fix bug in browse dataset --- mmseg/visualization/local_visualizer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmseg/visualization/local_visualizer.py b/mmseg/visualization/local_visualizer.py index ee3d652c7b..b9f2895cd2 100644 --- a/mmseg/visualization/local_visualizer.py +++ b/mmseg/visualization/local_visualizer.py @@ -136,6 +136,7 @@ def _draw_sem_seg(self, colors = [palette[label] for label in labels] mask = np.zeros_like(image, dtype=np.uint8) + mask = np.ascontiguousarray(mask) for label, color in zip(labels, colors): mask[sem_seg[0] == label, :] = color