提交 9277f8c4 编写于 作者: 文幕地方's avatar 文幕地方

update SLANet config

上级 a93367c9
...@@ -9,7 +9,7 @@ Global: ...@@ -9,7 +9,7 @@ Global:
eval_batch_step: [0, 1000] eval_batch_step: [0, 1000]
cal_metric_during_train: True cal_metric_during_train: True
pretrained_model: pretrained_model:
checkpoints: /ssd1/zhoujun20/table/ch/PaddleOCR/output/en/table_lcnet_1_0_csp_pan_headsv3_smooth_l1_pretrain_ssld_weight81_sync_bn/best_accuracy.pdparams checkpoints:
save_inference_dir: ./output/SLANet/infer save_inference_dir: ./output/SLANet/infer
use_visualdl: False use_visualdl: False
infer_img: doc/table/table.jpg infer_img: doc/table/table.jpg
...@@ -28,11 +28,7 @@ Optimizer: ...@@ -28,11 +28,7 @@ Optimizer:
beta2: 0.999 beta2: 0.999
clip_norm: 5.0 clip_norm: 5.0
lr: lr:
# name: Piecewise
learning_rate: 0.001 learning_rate: 0.001
# decay_epochs : [10, 20]
# values : [0.002, 0.0002, 0.0001]
# warmup_epoch: 0
regularizer: regularizer:
name: 'L2' name: 'L2'
factor: 0.00000 factor: 0.00000
...@@ -73,8 +69,8 @@ Metric: ...@@ -73,8 +69,8 @@ Metric:
Train: Train:
dataset: dataset:
name: PubTabDataSet name: PubTabDataSet
data_dir: /home/zhoujun20/table/PubTabNe/pubtabnet/train/ data_dir: train_data/table/pubtabnet/train/
label_file_list: [/home/zhoujun20/table/PubTabNe/pubtabnet/PubTabNet_2.0.0_train.jsonl] label_file_list: [train_data/table/pubtabnet/PubTabNet_2.0.0_train.jsonl]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
...@@ -109,7 +105,7 @@ Eval: ...@@ -109,7 +105,7 @@ Eval:
dataset: dataset:
name: PubTabDataSet name: PubTabDataSet
data_dir: /home/zhoujun20/table/PubTabNe/pubtabnet/val/ data_dir: /home/zhoujun20/table/PubTabNe/pubtabnet/val/
label_file_list: [/home/zhoujun20/table/PubTabNe/pubtabnet/PubTabNet_2.0.0_val.jsonl] label_file_list: [/home/zhoujun20/table/PubTabNe/pubtabnet/val_500.jsonl]
transforms: transforms:
- DecodeImage: # load image - DecodeImage: # load image
img_mode: BGR img_mode: BGR
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册