浏览代码

modify RGB to BGR in formula pipe

liuhongen1234567 1 年之前
父节点
当前提交
307765eeb7
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/pipelines/formula_recognition.py

+ 1 - 1
paddlex/inference/pipelines/formula_recognition.py

@@ -39,7 +39,7 @@ class FormulaRecognitionPipeline(BasePipeline):
             layout_batch_size=layout_batch_size,
             formula_rec_batch_size=formula_rec_batch_size,
         )
-        self.img_reader = ReadImage(format="RGB")
+        self.img_reader = ReadImage(format="BGR")
 
     def _build_predictor(self, layout_model, formula_rec_model):
         self.layout_predictor = self._create(model=layout_model)