Przeglądaj źródła

modify RGB to BGR in formula pipe

liuhongen1234567 1 rok temu
rodzic
commit
307765eeb7

+ 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)