Fix palette type. (#27)

2020-07-23 13:01:31 +08:00 · 2020-07-23 13:01:31 +08:00 · 579f5f883e
parent 1bedfe305b
commit 579f5f883e
1 changed files with 1 additions and 2 deletions
--- 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