modified config

pull/6852/head
smilelite 2022-07-11 23:28:51 +08:00
parent 6aa35c18ae
commit c53a6e5137
3 changed files with 7 additions and 5 deletions
test_tipc/configs/rec_r31_robustscanner

2
.gitignore vendored
View File

@ -10,6 +10,8 @@ __pycache__/
inference/
inference_results/
output/
train_data
log
*.DS_Store
*.vs

View File

@ -96,7 +96,7 @@ Eval:
- SARLabelEncode: # Class handling label
- RobustScannerRecResizeImg:
image_shape: [3, 48, 48, 160]
max_seq_len: *max_text_length
max_text_length: *max_text_length
width_downsample_ratio: 0.25
- KeepKeys:
keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order

View File

@ -12,7 +12,7 @@ Global:
checkpoints:
save_inference_dir:
use_visualdl: False
infer_img:
infer_img: ./inference/rec_inference
# for data or label process
character_dict_path: ppocr/utils/dict90.txt
max_text_length: &max_text_length 40
@ -49,7 +49,7 @@ Architecture:
mask: True
padding_idx: 92
encode_value: False
max_seq_len: 40
max_text_length: *max_text_length
Loss:
name: SARLoss
@ -75,7 +75,7 @@ Train:
- RobustScannerRecResizeImg:
image_shape: [3, 48, 48, 160] # h:48 w:[48,160]
width_downsample_ratio: 0.25
max_seq_len: *max_text_length
max_text_length: *max_text_length
- KeepKeys:
keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order
loader:
@ -97,7 +97,7 @@ Eval:
- SARLabelEncode: # Class handling label
- RobustScannerRecResizeImg:
image_shape: [3, 48, 48, 160]
max_seq_len: *max_text_length
max_text_length: *max_text_length
width_downsample_ratio: 0.25
- KeepKeys:
keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order