Jelajahi Sumber

add d2s_train_shapefor ocr recog module (#2586)

liuhongen1234567 11 bulan lalu
induk
melakukan
c5fc719dd2

+ 1 - 0
paddlex/repo_apis/PaddleOCR_api/configs/LaTeX_OCR_rec.yml

@@ -18,6 +18,7 @@ Global:
   use_space_char: False
   rec_char_dict_path:  ppocr/utils/dict/latex_ocr_tokenizer.json
   save_res_path: ./output/rec/predicts_latexocr.txt
+  d2s_train_image_shape: [1,256,256]
 
 Optimizer:
   name: AdamW

+ 1 - 1
paddlex/repo_apis/PaddleOCR_api/configs/PP-OCRv4_mobile_rec.yaml

@@ -19,7 +19,7 @@ Global:
   use_space_char: true
   distributed: true
   save_res_path: ./output/rec/predicts_ppocrv3.txt
-
+  d2s_train_image_shape: [3,48,320]
 
 Optimizer:
   name: Adam

+ 1 - 1
paddlex/repo_apis/PaddleOCR_api/configs/PP-OCRv4_server_rec.yaml

@@ -19,7 +19,7 @@ Global:
   use_space_char: true
   distributed: true
   save_res_path: ./output/rec/predicts_ppocrv3.txt
-
+  d2s_train_image_shape: [3,48,320]
 
 Optimizer:
   name: Adam

+ 1 - 0
paddlex/repo_apis/PaddleOCR_api/configs/SLANet.yaml

@@ -21,6 +21,7 @@ Global:
   infer_mode: False
   use_sync_bn: True
   save_res_path: output/infer
+  d2s_train_image_shape: [3,488,488]
 
 Optimizer:
   name: Adam

+ 1 - 0
paddlex/repo_apis/PaddleOCR_api/configs/SLANet_plus.yaml

@@ -21,6 +21,7 @@ Global:
   infer_mode: False
   use_sync_bn: True
   save_res_path: output/infer
+  d2s_train_image_shape: [3,488,488]
 
 Optimizer:
   name: Adam

+ 1 - 0
paddlex/repo_apis/PaddleOCR_api/configs/ch_RepSVTR_rec.yaml

@@ -19,6 +19,7 @@ Global:
   use_space_char: true
   distributed: true
   save_res_path: ./output/rec/predicts_repsvtr.txt
+  d2s_train_image_shape: [3,48,320]
 
 Optimizer:
   name: AdamW

+ 1 - 1
paddlex/repo_apis/PaddleOCR_api/configs/ch_SVTRv2_rec.yaml

@@ -19,7 +19,7 @@ Global:
   use_space_char: true
   distributed: true
   save_res_path: ./output/rec/predicts_svrtv2.txt
-
+  d2s_train_image_shape: [3,48,320]
 
 Optimizer:
   name: AdamW