update tablemv3 config
parent
f2c986d625
commit
fb9b01a152
|
@ -17,7 +17,7 @@ Global:
|
||||||
# for data or label process
|
# for data or label process
|
||||||
character_dict_path: ppocr/utils/dict/table_structure_dict.txt
|
character_dict_path: ppocr/utils/dict/table_structure_dict.txt
|
||||||
character_type: en
|
character_type: en
|
||||||
max_text_length: &max_text_length 800
|
max_text_length: &max_text_length 500
|
||||||
box_format: &box_format 'xyxy' # 'xywh', 'xyxy', 'xyxyxyxy'
|
box_format: &box_format 'xyxy' # 'xywh', 'xyxy', 'xyxyxyxy'
|
||||||
infer_mode: False
|
infer_mode: False
|
||||||
|
|
||||||
|
@ -38,7 +38,8 @@ Architecture:
|
||||||
Backbone:
|
Backbone:
|
||||||
name: MobileNetV3
|
name: MobileNetV3
|
||||||
scale: 1.0
|
scale: 1.0
|
||||||
model_name: large
|
model_name: small
|
||||||
|
disable_se: true
|
||||||
Head:
|
Head:
|
||||||
name: TableAttentionHead
|
name: TableAttentionHead
|
||||||
hidden_size: 256
|
hidden_size: 256
|
||||||
|
@ -89,7 +90,7 @@ Train:
|
||||||
keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ]
|
keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ]
|
||||||
loader:
|
loader:
|
||||||
shuffle: True
|
shuffle: True
|
||||||
batch_size_per_card: 32
|
batch_size_per_card: 48
|
||||||
drop_last: True
|
drop_last: True
|
||||||
num_workers: 1
|
num_workers: 1
|
||||||
|
|
||||||
|
@ -124,5 +125,5 @@ Eval:
|
||||||
loader:
|
loader:
|
||||||
shuffle: False
|
shuffle: False
|
||||||
drop_last: False
|
drop_last: False
|
||||||
batch_size_per_card: 16
|
batch_size_per_card: 48
|
||||||
num_workers: 1
|
num_workers: 1
|
||||||
|
|
Loading…
Reference in New Issue