未验证 提交 1e9fe08f 编写于 作者: 文幕地方's avatar 文幕地方 提交者: GitHub

Update README.md

上级 113c9bb5
...@@ -193,7 +193,7 @@ python3 eval_ser.py \ ...@@ -193,7 +193,7 @@ python3 eval_ser.py \
```shell ```shell
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3.7 infer_ser.py \ python3.7 infer_ser.py \
--model_name_or_path "./PP-Layout_v1.0_ser_pretrained/" \ --model_name_or_path "PP-Layout_v1.0_ser_pretrained/" \
--ser_model_type "LayoutXLM" \ --ser_model_type "LayoutXLM" \
--output_dir "output/ser/" \ --output_dir "output/ser/" \
--infer_imgs "XFUND/zh_val/image/" \ --infer_imgs "XFUND/zh_val/image/" \
...@@ -207,7 +207,7 @@ python3.7 infer_ser.py \ ...@@ -207,7 +207,7 @@ python3.7 infer_ser.py \
```shell ```shell
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3.7 infer_ser_e2e.py \ python3.7 infer_ser_e2e.py \
--model_name_or_path "./output/PP-Layout_v1.0_ser_pretrained/" \ --model_name_or_path "PP-Layout_v1.0_ser_pretrained/" \
--ser_model_type "LayoutXLM" \ --ser_model_type "LayoutXLM" \
--max_seq_length 512 \ --max_seq_length 512 \
--output_dir "output/ser_e2e/" \ --output_dir "output/ser_e2e/" \
...@@ -279,7 +279,7 @@ python3 train_re.py \ ...@@ -279,7 +279,7 @@ python3 train_re.py \
```shell ```shell
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3 eval_re.py \ python3 eval_re.py \
--model_name_or_path "output/check/checkpoint-best" \ --model_name_or_path "PP-Layout_v1.0_re_pretrained/" \
--max_seq_length 512 \ --max_seq_length 512 \
--eval_data_dir "XFUND/zh_val/image" \ --eval_data_dir "XFUND/zh_val/image" \
--eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \ --eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \
...@@ -297,7 +297,7 @@ python3 eval_re.py \ ...@@ -297,7 +297,7 @@ python3 eval_re.py \
```shell ```shell
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3 infer_re.py \ python3 infer_re.py \
--model_name_or_path "./PP-Layout_v1.0_re_pretrained/" \ --model_name_or_path "PP-Layout_v1.0_re_pretrained/" \
--max_seq_length 512 \ --max_seq_length 512 \
--eval_data_dir "XFUND/zh_val/image" \ --eval_data_dir "XFUND/zh_val/image" \
--eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \ --eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \
...@@ -314,8 +314,8 @@ python3 infer_re.py \ ...@@ -314,8 +314,8 @@ python3 infer_re.py \
```shell ```shell
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python3.7 infer_ser_re_e2e.py \ python3.7 infer_ser_re_e2e.py \
--model_name_or_path "./PP-Layout_v1.0_ser_pretrained/" \ --model_name_or_path "PP-Layout_v1.0_ser_pretrained/" \
--re_model_name_or_path "./PP-Layout_v1.0_re_pretrained/" \ --re_model_name_or_path "PP-Layout_v1.0_re_pretrained/" \
--ser_model_type "LayoutXLM" \ --ser_model_type "LayoutXLM" \
--max_seq_length 512 \ --max_seq_length 512 \
--output_dir "output/ser_re_e2e/" \ --output_dir "output/ser_re_e2e/" \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册