From a932cbe02ef06af2538e3fc2f5d0ed901ccee2a2 Mon Sep 17 00:00:00 2001
From: ycr <34575412+ClaireYiu@users.noreply.github.com>
Date: Thu, 23 Jul 2020 13:01:31 +0800
Subject: [PATCH] Fix palette type. (#27)

---
 mmseg/models/segmentors/base.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py
index 6b4084c77..6f59dbc72 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