浏览代码

fix_uvdoc_bgr (#3597)

Sunflower7788 8 月之前
父节点
当前提交
0151ca3e0c
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/models/image_unwarping/predictor.py

+ 1 - 1
paddlex/inference/models/image_unwarping/predictor.py

@@ -66,7 +66,7 @@ class WarpPredictor(BasicPredictor):
         Returns:
             tuple: A tuple containing the preprocessors, inference engine, and postprocessors.
         """
-        preprocessors = {"Read": ReadImage(format="RGB")}
+        preprocessors = {"Read": ReadImage(format="BGR")}
         preprocessors["Normalize"] = Normalize(mean=0.0, std=1.0, scale=1.0 / 255)
         preprocessors["ToCHW"] = ToCHWImage()
         preprocessors["ToBatch"] = ToBatch()