mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Fix palette type. (#27)

Browse Source
This commit is contained in:
ycr 2020-07-23 13:01:31 +08:00 committed by GitHub
parent 6e96bf83e5
commit a932cbe02e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmseg/models/segmentors/base.py
View File

@ -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