rodrigomasini commited on
Commit
2c8ba30
·
verified ·
1 Parent(s): 0112306

Update mdr_pdf_parser.py

Browse files
Files changed (1) hide show
  1. mdr_pdf_parser.py +4 -4
mdr_pdf_parser.py CHANGED
@@ -1876,7 +1876,7 @@ class _MDR_ONNXParams:
1876
 
1877
  # Attributes with default values (Group 1)
1878
  use_angle_cls: bool = True
1879
- rec_image_shape: str = "3,48,640"
1880
  cls_image_shape: str = "3,48,192"
1881
  cls_batch_num: int = 6
1882
  cls_thresh: float = 0.9
@@ -1934,9 +1934,9 @@ class MDROcrEngine:
1934
  rec_model_dir=paths["rec"],
1935
  rec_char_dict_path=paths["keys"],
1936
  # much lower thresholds so we actually get some candidate masks:
1937
- det_db_thresh=0.15,
1938
- det_db_box_thresh=0.15,
1939
- drop_score=0.01,
1940
  use_angle_cls=False,
1941
  )
1942
  try:
 
1876
 
1877
  # Attributes with default values (Group 1)
1878
  use_angle_cls: bool = True
1879
+ rec_image_shape: str = "3,48,256"
1880
  cls_image_shape: str = "3,48,192"
1881
  cls_batch_num: int = 6
1882
  cls_thresh: float = 0.9
 
1934
  rec_model_dir=paths["rec"],
1935
  rec_char_dict_path=paths["keys"],
1936
  # much lower thresholds so we actually get some candidate masks:
1937
+ det_db_thresh=0.3,
1938
+ det_db_box_thresh=0.5,
1939
+ drop_score=0.1,
1940
  use_angle_cls=False,
1941
  )
1942
  try: