diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py index 6b4084c77e..6f59dbc72e 100644 --- a/mmseg/models/segmentors/base.py +++ b/mmseg/models/segmentors/base.py @@ -242,8 +242,7 @@ def show_result(self, 0, 255, size=(len(self.CLASSES), 3)) else: palette = self.PALETTE - else: - palette = np.array(palette) + palette = np.array(palette) assert palette.shape[0] == len(self.CLASSES) assert palette.shape[1] == 3 assert len(palette.shape) == 2