Сделал отдельный конфиг для Kaggle.
Some checks are pending
Build/Publish Develop Docs / deploy (push) Waiting to run

This commit is contained in:
2025-07-21 19:03:52 +03:00
parent 206c2a6877
commit 5a87e37bb2
3 changed files with 145 additions and 9 deletions

View File

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