diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py index 6b4084c77e2cbd805adeec04744a6c32a4c8e27e..6f59dbc72ed91501aed9cab114cd13ebefea76e2 100644 --- a/mmseg/models/segmentors/base.py +++ b/mmseg/models/segmentors/base.py @@ -242,8 +242,7 @@ class BaseSegmentor(nn.Module): 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