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

fix: reduce recognition batch size for improved performance

myhloli преди 3 месеца
родител
ревизия
9ca1bf232b
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      mineru/model/ocr/paddleocr2pytorch/pytorch_paddle.py

+ 1 - 1
mineru/model/ocr/paddleocr2pytorch/pytorch_paddle.py

@@ -89,7 +89,7 @@ class PytorchPaddleOCR(TextSystem):
         kwargs['det_model_path'] = det_model_path
         kwargs['rec_model_path'] = rec_model_path
         kwargs['rec_char_dict_path'] = os.path.join(root_dir, 'pytorchocr', 'utils', 'resources', 'dict', dict_file)
-        kwargs['rec_batch_num'] = 16
+        kwargs['rec_batch_num'] = 8
 
         kwargs['device'] = device