Skip to content
Snippets Groups Projects
Commit a932cbe0 authored by ycr's avatar ycr Committed by GitHub
Browse files

Fix palette type. (#27)

parent 6e96bf83
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment