From e6c0508ff4ee2ca4c33762a9ffb9eb2d46a341de Mon Sep 17 00:00:00 2001 From: wuzekai <3025054974@qq.com> Date: Tue, 8 Jul 2025 05:31:32 +0000 Subject: [PATCH] Delete ch_det_res18_db_v2.0.yml --- config/ch_det_res18_db_v2.0.yml | 132 -------------------------------- 1 file changed, 132 deletions(-) delete mode 100644 config/ch_det_res18_db_v2.0.yml diff --git a/config/ch_det_res18_db_v2.0.yml b/config/ch_det_res18_db_v2.0.yml deleted file mode 100644 index 084e6e6..0000000 --- a/config/ch_det_res18_db_v2.0.yml +++ /dev/null @@ -1,132 +0,0 @@ -Global: - use_gpu: true - epoch_num: 250 - log_smooth_window: 20 - print_batch_step: 30 - save_model_dir: ./output/ch_db_res18/ - save_epoch_step: 500 - # evaluation is run every 5000 iterations after the 4000th iteration - eval_batch_step: 60 - cal_metric_during_train: False - pretrained_model: ./pretrain_models/ch_ppocr_server_v2.0_det_train/best_accuracy.pdparams - checkpoints: - save_inference_dir: - use_visualdl: false - use_wandb: true - infer_img: doc/imgs_en/img_10.jpg - save_res_path: ./output/test/det_db/predicts_db.txt - name: ch_db_res18 - -Architecture: - model_type: det - algorithm: DB - Transform: - Backbone: - name: ResNet_vd - layers: 18 - disable_se: True - Neck: - name: DBFPN - out_channels: 256 - Head: - name: DBHead - k: 50 - -Loss: - name: DBLoss - balance_loss: true - main_loss_type: DiceLoss - alpha: 5 - beta: 10 - ohem_ratio: 3 - -Optimizer: - name: Adam - beta1: 0.9 - beta2: 0.999 - lr: - name: Cosine - learning_rate: 0.001 - warmup_epoch: 2 - regularizer: - name: 'L2' - factor: 0 - -PostProcess: - name: DBPostProcess - thresh: 0.3 - box_thresh: 0.6 - max_candidates: 1000 - unclip_ratio: 1.5 - -Metric: - name: DetMetric - main_indicator: hmean - -Train: - dataset: - name: SimpleDataSet - data_dir: ./train_data/ - label_file_list: - - ./train_data/det/train.txt - ratio_list: [1.0] - transforms: - - DecodeImage: # load image - img_mode: BGR - channel_first: False - - DetLabelEncode: # Class handling label - - IaaAugment: - augmenter_args: - - { 'type': Affine, 'args': { 'rotate': [-3, 3] } } # 旋转 - - { 'type': Resize, 'args': { 'size': [0.8, 1.2] } } #缩放 - - EastRandomCropData: - size: [960, 960] - max_tries: 50 - keep_ratio: true - - MakeBorderMap: - shrink_ratio: 0.4 - thresh_min: 0.3 - thresh_max: 0.7 - - MakeShrinkMap: - shrink_ratio: 0.4 - min_text_size: 8 - - NormalizeImage: - scale: 1./255. - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: 'hwc' - - ToCHWImage: - - KeepKeys: - keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list - loader: - shuffle: True - drop_last: False - batch_size_per_card: 8 - num_workers: 4 - -Eval: - dataset: - name: SimpleDataSet - data_dir: ./train_data/ - label_file_list: - - ./train_data/det/val.txt - transforms: - - DecodeImage: # load image - img_mode: BGR - channel_first: False - - DetLabelEncode: # Class handling label - - DetResizeForTest: -# image_shape: [736, 1280] - - NormalizeImage: - scale: 1./255. - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: 'hwc' - - ToCHWImage: - - KeepKeys: - keep_keys: ['image', 'shape', 'polys', 'ignore_tags'] - loader: - shuffle: False - drop_last: False - batch_size_per_card: 1 # must be 1 - num_workers: 2 -- 2.22.0