diff --git a/configs/rec_custom.yml b/configs/rec_custom.yml index 7e467cf..0dc37a6 100644 --- a/configs/rec_custom.yml +++ b/configs/rec_custom.yml @@ -5,7 +5,7 @@ Global: epoch_num: 75 log_smooth_window: 20 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 eval_batch_step: [0, 2000] cal_metric_during_train: true @@ -14,13 +14,13 @@ Global: checkpoints: save_inference_dir: use_visualdl: false - infer_img: ./input/custom-ocr-dataset/images/1C4HJXEN3MW645094.jpg - character_dict_path: ./input/custom-ocr-dataset/dict.txt + infer_img: /kaggle/input/custom-ocr-dataset/images/1C4HJXEN3MW645094.jpg + character_dict_path: /kaggle/input/custom-ocr-dataset/dict.txt max_text_length: &max_text_length 19 infer_mode: false use_space_char: 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] @@ -78,10 +78,10 @@ Train: dataset: name: MultiScaleDataSet ds_width: false - data_dir: ./input/custom-ocr-dataset/images + data_dir: /kaggle/input/custom-ocr-dataset/images ext_op_transform_idx: 1 label_file_list: - - ./input/custom-ocr-dataset/train.txt + - /kaggle/input/custom-ocr-dataset/train.txt transforms: - DecodeImage: img_mode: BGR @@ -111,9 +111,9 @@ Train: Eval: dataset: name: SimpleDataSet - data_dir: ./input/custom-ocr-dataset/images + data_dir: /kaggle/input/custom-ocr-dataset/images label_file_list: - - ./input/custom-ocr-dataset/val.txt + - /kaggle/input/custom-ocr-dataset/val.txt transforms: - DecodeImage: img_mode: BGR