update_det_static

Sunting78 · Sunting78 · commit e8576cbf35c1 · 2024-12-11T12:18:36.000Z
diff --git a/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_cml.yml b/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_cml.yml
@@ -16,6 +16,7 @@ Global:
   use_visualdl: false
   infer_img: doc/imgs_en/img_10.jpg
   save_res_path: ./checkpoints/det_db/predicts_db.txt
+  d2s_train_image_shape: [3, 640, 640]
   distributed: true
 Architecture:
   name: DistillationModel
diff --git a/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml b/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml
@@ -16,6 +16,7 @@ Global:
   use_visualdl: false
   infer_img: doc/imgs_en/img_10.jpg
   save_res_path: ./checkpoints/det_db/predicts_db.txt
+  d2s_train_image_shape: [3, 640, 640]
   distributed: true
 
 Architecture:
diff --git a/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml b/configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml
@@ -16,6 +16,7 @@ Global:
   use_visualdl: false
   infer_img: doc/imgs_en/img_10.jpg
   save_res_path: ./checkpoints/det_db/predicts_db.txt
+  d2s_train_image_shape: [3, 640, 640]
   distributed: true
 
 Architecture:
