Skip to content

Commit 492b040

Browse files
authored
Update layoutxlm (PaddlePaddle#1318)
* update layoutxlm * update layoutxlm
1 parent 471a8a0 commit 492b040

File tree

4 files changed

+17
-14
lines changed

4 files changed

+17
-14
lines changed

examples/multimodal/layoutxlm/README.md

+7-4
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,20 @@
88
### 配置环境
99
环境依赖
1010
- cv2
11-
- numpy
11+
- sentencepiece
12+
- yacs
1213

1314
安装命令:
1415
```shell
1516
pip install opencv-python
16-
pip install numpy
17+
pip install sentencepiece
18+
pip install yacs
1719
```
1820

1921
### 数据准备
20-
处理好的XFUND中文数据集下载地址:https://bj.bcebos.com/v1/paddlenlp/datasets/XFUND.zip
22+
处理好的XFUND中文数据集下载地址:https://bj.bcebos.com/v1/paddlenlp/datasets/XFUND.zip
23+
24+
下载并解压该数据集,解压后将数据集放置在当前目录下。
2125

2226
### 执行Fine-tuning
2327
1. ``Semantic Entity Recognition`` 任务启动Fine-tuning的方式如下:
@@ -28,7 +32,6 @@ pip install numpy
2832
# best metrics: {'precision': 0.8514686248331108, 'recall': 0.9354602126879354, 'f1': 0.8914904770225406}
2933
```
3034

31-
3235
2. ``Relation Extraction`` 任务启动Fine-tuning的方式如下:
3336
```shell
3437
bash run_xfun_re.sh

examples/multimodal/layoutxlm/run_xfun_re.sh

+4-4
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@ export CUDA_VISIBLE_DEVICES=0
33
python ./run_xfun_re.py \
44
--model_name_or_path "layoutxlm-base-uncased" \
55
--max_seq_length 512 \
6-
--train_data_dir "zh.train/img" \
7-
--train_label_path "zh.train/xfun_normalize_train.json" \
8-
--eval_data_dir "zh.val/img" \
9-
--eval_label_path "zh.val/xfun_normalize_val.json" \
6+
--train_data_dir "XFUND/zh_train/image" \
7+
--train_label_path "XFUND/zh_train/xfun_normalize_train.json" \
8+
--eval_data_dir "XFUND/zh_val/image" \
9+
--eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \
1010
--num_train_epochs 200 \
1111
--eval_steps 50 \
1212
--save_steps 500 \

examples/multimodal/layoutxlm/run_xfun_ser.sh

+4-4
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@ export CUDA_VISIBLE_DEVICES=0
33
python ./run_xfun_ser.py \
44
--model_name_or_path "layoutxlm-base-uncased" \
55
--max_seq_length 512 \
6-
--train_data_dir "zh.train/img" \
7-
--train_label_path "zh.train/xfun_normalize_train.json" \
8-
--eval_data_dir "zh.val/img" \
9-
--eval_label_path "zh.val/xfun_normalize_val.json" \
6+
--train_data_dir "XFUND/zh_train/image" \
7+
--train_label_path "XFUND/zh_train/xfun_normalize_train.json" \
8+
--eval_data_dir "XFUND/zh_val/image" \
9+
--eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \
1010
--num_train_epochs 200 \
1111
--eval_steps 10 \
1212
--save_steps 500 \

paddlenlp/transformers/layoutxlm/visual_backbone.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ def read_config(fp=None):
3232
dir_name = os.path.dirname(os.path.abspath(__file__))
3333
fp = os.path.join(dir_name, "visual_backbone.yaml")
3434
with open(fp, "r") as fin:
35-
yacs = try_import("yacs")
36-
cfg = yacs.config.CfgNode().load_cfg(fin)
35+
yacs_config = try_import("yacs.config")
36+
cfg = yacs_config.CfgNode().load_cfg(fin)
3737
cfg.freeze()
3838
return cfg
3939

0 commit comments

Comments
 (0)