Bläddra i källkod

fix BGR to RGB in ReadImage of seg predictor

gaotingquan 1 år sedan
förälder
incheckning
f3a6f8db70
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      paddlex/modules/semantic_segmentation/predictor/predictor.py

+ 1 - 1
paddlex/modules/semantic_segmentation/predictor/predictor.py

@@ -88,7 +88,7 @@ class SegPredictor(BasePredictor):
             f"Transformation operators for data preprocessing will be inferred from config file."
         )
         pre_transforms = self.other_src.pre_transforms
-        pre_transforms.insert(0, image_common.ReadImage())
+        pre_transforms.insert(0, image_common.ReadImage(format="RGB"))
         pre_transforms.append(image_common.ToCHWImage())
         return pre_transforms