@@ -34,6 +34,8 @@ class SegResult(CVResult):
"""apply"""
seg_map = self["pred"]
pc_map = self.get_pseudo_color_map(seg_map[0])
+ if pc_map.mode == "P":
+ pc_map = pc_map.convert("RGB")
return pc_map
def get_pseudo_color_map(self, pred):