Sfoglia il codice sorgente

Reduce recognition batch size from 8 to 6 for improved OCR performance

myhloli 1 mese fa
parent
commit
b7e4ea0b49
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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'] = 8
+        kwargs['rec_batch_num'] = 6
 
         kwargs['device'] = device