Selaa lähdekoodia

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

Zhang Zelun 10 kuukautta sitten
vanhempi
commit
56d23c77db
1 muutettua tiedostoa jossa 2 lisäystä ja 0 poistoa
  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):