Преглед на файлове

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,
             layout_batch_size=layout_batch_size,
             formula_rec_batch_size=formula_rec_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):
     def _build_predictor(self, layout_model, formula_rec_model):
         self.layout_predictor = self._create(model=layout_model)
         self.layout_predictor = self._create(model=layout_model)