Конфигурация для Kaggle.
Some checks are pending
Build/Publish Develop Docs / deploy (push) Waiting to run

This commit is contained in:
2025-07-08 11:19:39 +03:00
parent 12886be170
commit 6b31c54a26

View File

@@ -5,7 +5,7 @@ Global:
epoch_num: 75 epoch_num: 75
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: ./output/PP-OCRv5_server_rec save_model_dir: /kaggle/output/PP-OCRv5_server_rec
save_epoch_step: 1 save_epoch_step: 1
eval_batch_step: [0, 2000] eval_batch_step: [0, 2000]
cal_metric_during_train: true cal_metric_during_train: true
@@ -14,13 +14,13 @@ Global:
checkpoints: checkpoints:
save_inference_dir: save_inference_dir:
use_visualdl: false use_visualdl: false
infer_img: ./input/custom-ocr-dataset/images/1C4HJXEN3MW645094.jpg infer_img: /kaggle/input/custom-ocr-dataset/images/1C4HJXEN3MW645094.jpg
character_dict_path: ./input/custom-ocr-dataset/dict.txt character_dict_path: /kaggle/input/custom-ocr-dataset/dict.txt
max_text_length: &max_text_length 19 max_text_length: &max_text_length 19
infer_mode: false infer_mode: false
use_space_char: true use_space_char: true
distributed: true distributed: true
save_res_path: ./output/rec/predicts_ppocrv5.txt save_res_path: /kaggle/output/rec/predicts_ppocrv5.txt
d2s_train_image_shape: [3, 48, 320] d2s_train_image_shape: [3, 48, 320]
@@ -78,10 +78,10 @@ Train:
dataset: dataset:
name: MultiScaleDataSet name: MultiScaleDataSet
ds_width: false ds_width: false
data_dir: ./input/custom-ocr-dataset/images data_dir: /kaggle/input/custom-ocr-dataset/images
ext_op_transform_idx: 1 ext_op_transform_idx: 1
label_file_list: label_file_list:
- ./input/custom-ocr-dataset/train.txt - /kaggle/input/custom-ocr-dataset/train.txt
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR
@@ -111,9 +111,9 @@ Train:
Eval: Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./input/custom-ocr-dataset/images data_dir: /kaggle/input/custom-ocr-dataset/images
label_file_list: label_file_list:
- ./input/custom-ocr-dataset/val.txt - /kaggle/input/custom-ocr-dataset/val.txt
transforms: transforms:
- DecodeImage: - DecodeImage:
img_mode: BGR img_mode: BGR