Parcourir la source

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

Zhang Zelun il y a 10 mois
Parent
commit
56d23c77db
1 fichiers modifiés avec 2 ajouts et 0 suppressions
  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):