Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
a5dbd8bb
Commit
a5dbd8bb
authored
Dec 18, 2021
by
WenmuZhou
Browse files
rm unsused code
parent
5e7ad64c
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
12 deletions
+2
-12
ppstructure/vqa/README.md
ppstructure/vqa/README.md
+1
-11
ppstructure/vqa/train_ser.py
ppstructure/vqa/train_ser.py
+1
-1
No files found.
ppstructure/vqa/README.md
View file @
a5dbd8bb
...
@@ -182,16 +182,6 @@ python3 eval_ser.py \
...
@@ -182,16 +182,6 @@ python3 eval_ser.py \
```
```
最终会打印出
`precision`
,
`recall`
,
`f1`
等指标
最终会打印出
`precision`
,
`recall`
,
`f1`
等指标
```
shell
export
CUDA_VISIBLE_DEVICES
=
0
python3.7 infer_ser.py
\
--model_name_or_path
"./PP-Layout_v1.0_ser_pretrained/"
\
--output_dir
"output_res/"
\
--infer_imgs
"XFUND/zh_val/image/"
\
--ocr_json_path
"XFUND/zh_val/xfun_normalize_val.json"
```
*
使用评估集合中提供的OCR识别结果进行预测
*
使用评估集合中提供的OCR识别结果进行预测
```
shell
```
shell
...
@@ -237,7 +227,7 @@ python3 train_re.py \
...
@@ -237,7 +227,7 @@ python3 train_re.py \
--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"
\
--label_map_path
'labels/labels_ser.txt'
\
--label_map_path
'labels/labels_ser.txt'
\
--num_train_epochs
2
\
--num_train_epochs
2
00
\
--eval_steps
10
\
--eval_steps
10
\
--output_dir
"output/re/"
\
--output_dir
"output/re/"
\
--learning_rate
5e-5
\
--learning_rate
5e-5
\
...
...
ppstructure/vqa/train_ser.py
View file @
a5dbd8bb
...
@@ -139,7 +139,7 @@ def train(args):
...
@@ -139,7 +139,7 @@ def train(args):
global_step
=
0
global_step
=
0
tr_loss
=
0.0
tr_loss
=
0.0
set_seed
(
ags
.
seed
)
set_seed
(
a
r
gs
.
seed
)
best_metrics
=
None
best_metrics
=
None
train_reader_cost
=
0.0
train_reader_cost
=
0.0
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment