|
|
@@ -0,0 +1,169 @@
|
|
|
+Global:
|
|
|
+ debug: false
|
|
|
+ use_gpu: true
|
|
|
+ epoch_num: 100
|
|
|
+ log_smooth_window: 20
|
|
|
+ print_batch_step: 20
|
|
|
+ save_model_dir: ./output/ch_PP-OCR_V3_det_back_0213/
|
|
|
+ save_epoch_step: 100
|
|
|
+ eval_batch_step:
|
|
|
+ - 0
|
|
|
+ - 100
|
|
|
+ cal_metric_during_train: false
|
|
|
+ pretrained_model: pretrain_model/ch_PP-OCRv3_det_distill_train/student.pdparams
|
|
|
+ checkpoints: null
|
|
|
+ save_inference_dir: null
|
|
|
+ use_visualdl: true
|
|
|
+ infer_img: images/tw_01.png
|
|
|
+ save_res_path: ./infer_result/predicts_db.txt
|
|
|
+ distributed: true
|
|
|
+ label_list: configs/det/ch_PP-OCRv3/label_list.txt
|
|
|
+ num_classes: 12
|
|
|
+
|
|
|
+Architecture:
|
|
|
+ model_type: det
|
|
|
+ algorithm: DB
|
|
|
+ Transform:
|
|
|
+ Backbone:
|
|
|
+ name: MobileNetV3
|
|
|
+ scale: 1.25
|
|
|
+ model_name: large
|
|
|
+ disable_se: True
|
|
|
+ Neck:
|
|
|
+ name: RSEFPN
|
|
|
+ out_channels: 96
|
|
|
+ shortcut: True
|
|
|
+ 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: 5.0e-05
|
|
|
+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: /home/aistudio/data/data190513
|
|
|
+ # 数据集为tw_idcard_det_back_0210
|
|
|
+ label_file_list:
|
|
|
+ - /home/aistudio/data/data190513/Label.txt
|
|
|
+ ratio_list: [0.8]
|
|
|
+ transforms:
|
|
|
+ - DecodeImage:
|
|
|
+ img_mode: BGR
|
|
|
+ channel_first: false
|
|
|
+ - DetLabelEncode: null
|
|
|
+ - IaaAugment:
|
|
|
+ augmenter_args:
|
|
|
+ - type: Fliplr
|
|
|
+ args:
|
|
|
+ p: 0.5
|
|
|
+ - type: Affine
|
|
|
+ args:
|
|
|
+ rotate:
|
|
|
+ - -10
|
|
|
+ - 10
|
|
|
+ - type: Resize
|
|
|
+ args:
|
|
|
+ size:
|
|
|
+ - 0.5
|
|
|
+ - 3
|
|
|
+ - 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: null
|
|
|
+ - KeepKeys:
|
|
|
+ keep_keys:
|
|
|
+ - image
|
|
|
+ - threshold_map
|
|
|
+ - threshold_mask
|
|
|
+ - shrink_map
|
|
|
+ - shrink_mask
|
|
|
+ - class_mask
|
|
|
+ loader:
|
|
|
+ shuffle: true
|
|
|
+ drop_last: false
|
|
|
+ batch_size_per_card: 8
|
|
|
+ num_workers: 2
|
|
|
+Eval:
|
|
|
+ dataset:
|
|
|
+ name: SimpleDataSet
|
|
|
+ data_dir: /home/aistudio/data/data190513
|
|
|
+ label_file_list:
|
|
|
+ - /home/aistudio/data/data190513/Label.txt
|
|
|
+ ratio_list: [0.2]
|
|
|
+ transforms:
|
|
|
+ - DecodeImage:
|
|
|
+ img_mode: BGR
|
|
|
+ channel_first: false
|
|
|
+ - DetLabelEncode: null
|
|
|
+ - DetResizeForTest:
|
|
|
+ resize_long: 960
|
|
|
+ - NormalizeImage:
|
|
|
+ scale: 1./255.
|
|
|
+ mean:
|
|
|
+ - 0.485
|
|
|
+ - 0.456
|
|
|
+ - 0.406
|
|
|
+ std:
|
|
|
+ - 0.229
|
|
|
+ - 0.224
|
|
|
+ - 0.225
|
|
|
+ order: hwc
|
|
|
+ - ToCHWImage: null
|
|
|
+ - KeepKeys:
|
|
|
+ keep_keys:
|
|
|
+ - image
|
|
|
+ - shape
|
|
|
+ - polys
|
|
|
+ - ignore_tags
|
|
|
+ loader:
|
|
|
+ shuffle: false
|
|
|
+ drop_last: false
|
|
|
+ batch_size_per_card: 1
|
|
|
+ num_workers: 2
|
