Prechádzať zdrojové kódy

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

Zhang Zelun 10 mesiacov pred
rodič
commit
56d23c77db
1 zmenil súbory, kde vykonal 2 pridanie a 0 odobranie
  1. 2 0
      paddlex/inference/results/seg.py

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