Эх сурвалжийг харах

adapt seg rst to lower version of pillow (#2757)

Zhang Zelun 10 сар өмнө
parent
commit
56d23c77db

+ 2 - 0
paddlex/inference/results/seg.py

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