提交 7c96520d 编写于 作者: W WenmuZhou

yml文件去除个人路径

上级 388d8dae
...@@ -3,15 +3,15 @@ Global: ...@@ -3,15 +3,15 @@ Global:
epoch_num: 1200 epoch_num: 1200
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 2 print_batch_step: 2
save_model_dir: ./output/20201015_r50/ save_model_dir: ./output/det_r50_vd/
save_epoch_step: 1200 save_epoch_step: 1200
# evaluation is run every 5000 iterations after the 4000th iteration # evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step: 8 eval_batch_step: 8
# if pretrained_model is saved in static mode, load_static_weights must set to True # if pretrained_model is saved in static mode, load_static_weights must set to True
load_static_weights: True load_static_weights: True
cal_metric_during_train: False cal_metric_during_train: False
pretrained_model: /home/zhoujun20/pretrain_models/ResNet50_vd_ssld_pretrained/ pretrained_model: ./pretrain_models/ResNet50_vd_ssld_pretrained/
checkpoints: #./output/det_db_0.001_DiceLoss_256_pp_config_2.0b_4gpu/best_accuracy checkpoints:
save_inference_dir: save_inference_dir:
use_visualdl: True use_visualdl: True
infer_img: doc/imgs_en/img_10.jpg infer_img: doc/imgs_en/img_10.jpg
...@@ -65,9 +65,9 @@ Metric: ...@@ -65,9 +65,9 @@ Metric:
TRAIN: TRAIN:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: /home/zhoujun20/detection/ data_dir: ./detection/
file_list: file_list:
- /home/zhoujun20/detection/train_icdar2015_label.txt # dataset1 - ./detection/train_icdar2015_label.txt # dataset1
ratio_list: [1.0] ratio_list: [1.0]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
...@@ -107,9 +107,9 @@ TRAIN: ...@@ -107,9 +107,9 @@ TRAIN:
EVAL: EVAL:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: /home/zhoujun20/detection/ data_dir: ./detection/
file_list: file_list:
- /home/zhoujun20/detection/test_icdar2015_label.txt - ./detection/test_icdar2015_label.txt
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
......
...@@ -68,7 +68,7 @@ TRAIN: ...@@ -68,7 +68,7 @@ TRAIN:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./rec data_dir: ./rec
file_list: file_list:
- ./rec/real_data.txt # dataset1 - ./rec/train.txt # dataset1
ratio_list: [ 0.4,0.6 ] ratio_list: [ 0.4,0.6 ]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
...@@ -91,7 +91,7 @@ EVAL: ...@@ -91,7 +91,7 @@ EVAL:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./rec data_dir: ./rec
file_list: file_list:
- ./rec/label_val_all.txt - ./rec/val.txt
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
......
Global: Global:
use_gpu: false use_gpu: false
epoch_num: 500 epoch_num: 72
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 1 print_batch_step: 10
save_model_dir: ./output/rec/test/ save_model_dir: ./output/rec/mv3_none_none_ctc/
save_epoch_step: 500 save_epoch_step: 500
# evaluation is run every 5000 iterations after the 4000th iteration # evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step: 1016 eval_batch_step: 2000
# if pretrained_model is saved in static mode, load_static_weights must set to True # if pretrained_model is saved in static mode, load_static_weights must set to True
load_static_weights: True load_static_weights: True
cal_metric_during_train: True cal_metric_during_train: True
pretrained_model: pretrained_model:
checkpoints: #output/rec/rec_crnn/best_accuracy checkpoints:
save_inference_dir: save_inference_dir:
use_visualdl: True use_visualdl: True
infer_img: doc/imgs_words/ch/word_1.jpg infer_img: doc/imgs_words/ch/word_1.jpg
# for data or label process # for data or label process
max_text_length: 80 max_text_length: 25
character_dict_path: /home/zhoujun20/rec/lmdb/dict.txt character_dict_path:
character_type: 'en' character_type: 'en'
use_space_char: True use_space_char: False
infer_mode: False infer_mode: False
use_tps: False use_tps: False
...@@ -29,9 +29,9 @@ Optimizer: ...@@ -29,9 +29,9 @@ Optimizer:
beta1: 0.9 beta1: 0.9
beta2: 0.999 beta2: 0.999
learning_rate: learning_rate:
name: Cosine # name: Cosine
lr: 0.0005 lr: 0.0005
warmup_epoch: 1 # warmup_epoch: 1
regularizer: regularizer:
name: 'L2' name: 'L2'
factor: 0.00001 factor: 0.00001
...@@ -43,7 +43,7 @@ Architecture: ...@@ -43,7 +43,7 @@ Architecture:
Backbone: Backbone:
name: MobileNetV3 name: MobileNetV3
scale: 0.5 scale: 0.5
model_name: small model_name: large
small_stride: [ 1, 2, 2, 2 ] small_stride: [ 1, 2, 2, 2 ]
Neck: Neck:
name: SequenceEncoder name: SequenceEncoder
...@@ -66,7 +66,7 @@ TRAIN: ...@@ -66,7 +66,7 @@ TRAIN:
dataset: dataset:
name: LMDBDateSet name: LMDBDateSet
file_list: file_list:
- /Users/zhoujun20/Downloads/evaluation_new # dataset1 - ./rec/train # dataset1
ratio_list: [ 0.4,0.6 ] ratio_list: [ 0.4,0.6 ]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
...@@ -75,7 +75,7 @@ TRAIN: ...@@ -75,7 +75,7 @@ TRAIN:
- CTCLabelEncode: # Class handling label - CTCLabelEncode: # Class handling label
- RecAug: - RecAug:
- RecResizeImg: - RecResizeImg:
image_shape: [ 3,32,320 ] image_shape: [ 3,32,100 ]
- keepKeys: - keepKeys:
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
loader: loader:
...@@ -88,14 +88,14 @@ EVAL: ...@@ -88,14 +88,14 @@ EVAL:
dataset: dataset:
name: LMDBDateSet name: LMDBDateSet
file_list: file_list:
- /home/zhoujun20/rec/lmdb/val - ./rec/val/
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
channel_first: False channel_first: False
- CTCLabelEncode: # Class handling label - CTCLabelEncode: # Class handling label
- RecResizeImg: - RecResizeImg:
image_shape: [ 3,32,320 ] image_shape: [ 3,32,100 ]
- keepKeys: - keepKeys:
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
loader: loader:
......
...@@ -64,9 +64,9 @@ Metric: ...@@ -64,9 +64,9 @@ Metric:
TRAIN: TRAIN:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: /home/zhoujun20/rec data_dir: ./rec
file_list: file_list:
- /home/zhoujun20/rec/real_data.txt # dataset1 - ./rec/train.txt # dataset1
ratio_list: [ 0.4,0.6 ] ratio_list: [ 0.4,0.6 ]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
...@@ -87,9 +87,9 @@ TRAIN: ...@@ -87,9 +87,9 @@ TRAIN:
EVAL: EVAL:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: /home/zhoujun20/rec data_dir: ./rec
file_list: file_list:
- /home/zhoujun20/rec/label_val_all.txt - ./rec/val.txt
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
......
Global:
use_gpu: false
epoch_num: 500
log_smooth_window: 20
print_batch_step: 10
save_model_dir: ./output/rec/res34_none_none_ctc/
save_epoch_step: 500
# evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step: 127
# if pretrained_model is saved in static mode, load_static_weights must set to True
load_static_weights: True
cal_metric_during_train: True
pretrained_model:
checkpoints:
save_inference_dir:
use_visualdl: False
infer_img: doc/imgs_words/ch/word_1.jpg
# for data or label process
max_text_length: 80
character_dict_path: ppocr/utils/ppocr_keys_v1.txt
character_type: 'ch'
use_space_char: False
infer_mode: False
use_tps: False
Optimizer:
name: Adam
beta1: 0.9
beta2: 0.999
learning_rate:
name: Cosine
lr: 0.001
warmup_epoch: 4
regularizer:
name: 'L2'
factor: 0.00001
Architecture:
type: rec
algorithm: CRNN
Transform:
Backbone:
name: ResNet
layers: 34
Neck:
name: SequenceEncoder
encoder_type: reshape
Head:
name: CTC
fc_decay: 0.00001
Loss:
name: CTCLoss
PostProcess:
name: CTCLabelDecode
Metric:
name: RecMetric
main_indicator: acc
TRAIN:
dataset:
name: SimpleDataSet
data_dir: ./rec
file_list:
- ./rec/train.txt # dataset1
ratio_list: [ 0.4,0.6 ]
transforms:
- DecodeImage: # load image
img_mode: BGR
channel_first: False
- CTCLabelEncode: # Class handling label
- RecAug:
- RecResizeImg:
image_shape: [ 3,32,320 ]
- keepKeys:
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
loader:
batch_size: 256
shuffle: True
drop_last: True
num_workers: 8
EVAL:
dataset:
name: SimpleDataSet
data_dir: ./rec
file_list:
- ./rec/val.txt
transforms:
- DecodeImage: # load image
img_mode: BGR
channel_first: False
- CTCLabelEncode: # Class handling label
- RecResizeImg:
image_shape: [ 3,32,320 ]
- keepKeys:
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
loader:
shuffle: False
drop_last: False
batch_size: 256
num_workers: 8
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册