Эх сурвалжийг харах

Merge pull request #3700 from myhloli/ocr_enhance

Reduce recognition batch size from 8 to 6
Xiaomeng Zhao 1 сар өмнө
parent
commit
aa4bc6259b

+ 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