Ver Fonte

fix static train in formula (#3558)

liuhongen1234567 há 8 meses atrás
pai
commit
6f031fcf4d

+ 2 - 1
paddlex/repo_apis/PaddleOCR_api/configs/PP-FormulaNet-L.yaml

@@ -18,9 +18,10 @@ Global:
   rec_char_dict_path: &rec_char_dict_path ppocr/utils/dict/unimernet_tokenizer
   max_new_tokens: &max_new_tokens 1024
   input_size: &input_size [768, 768]
-  save_res_path: ./output/rec/predicts_unimernet_latexocr.txt
+  save_res_path: ./output/rec/predicts_pp_formulanet_l.txt
   allow_resize_largeImg: False
   start_ema: True
+  d2s_train_image_shape: [1,768,768]
 
 Optimizer:
   name: AdamW

+ 2 - 1
paddlex/repo_apis/PaddleOCR_api/configs/PP-FormulaNet-S.yaml

@@ -18,9 +18,10 @@ Global:
   rec_char_dict_path: &rec_char_dict_path  ppocr/utils/dict/unimernet_tokenizer
   max_new_tokens: &max_new_tokens 1024
   input_size: &input_size [384, 384]
-  save_res_path: ./output/rec/predicts_unimernet_latexocr.txt
+  save_res_path: ./output/rec/predicts_pp_formulanet_s.txt
   allow_resize_largeImg: False
   start_ema: True
+  d2s_train_image_shape: [1,384,384]
 
 Optimizer:
   name: AdamW

+ 2 - 1
paddlex/repo_apis/PaddleOCR_api/configs/UniMERNet.yaml

@@ -18,8 +18,9 @@ Global:
   rec_char_dict_path: &rec_char_dict_path ppocr/utils/dict/unimernet_tokenizer
   input_size: &input_size [192, 672]
   max_seq_len: &max_seq_len 1024
-  save_res_path: ./output/rec/predicts_unimernet_plus_config_latexocr.txt
+  save_res_path: ./output/rec/predicts_unimernet.txt
   allow_resize_largeImg: False
+  d2s_train_image_shape: [1,192,672]
 
 Optimizer:
   name: AdamW