Просмотр исходного кода

fix: reduce recognition batch size for improved performance

myhloli 3 месяцев назад
Родитель
Сommit
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