forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

rename key

This commit is contained in:
LDOUBLEV 2021-06-08 16:11:20 +08:00
parent 27b543ab13
commit 40f78f7531
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/params.txt
View File

@ -1,4 +1,4 @@
train_model_list: det;benchmark/benchmark_det.yml train_model_list: ocr_det
gpu_list: -1|0|0,1 gpu_list: -1|0|0,1
auto_cast_list: False|True auto_cast_list: False|True
trainer_list: norm|quant|prune trainer_list: norm|quant|prune

4
test/test.sh
View File

@ -55,10 +55,10 @@ checkpoints=$(func_parser "${lines[14]}")
for train_model in ${train_model_list[*]}; do for train_model in ${train_model_list[*]}; do
if [ ${train_model} = "det" ];then if [ ${train_model} = "ocr_det" ];then
model_name="det" model_name="det"
yml_file="configs/det/det_mv3_db.yml" yml_file="configs/det/det_mv3_db.yml"
elif [ ${train_model} = "rec" ];then elif [ ${train_model} = "ocr_rec" ];then
model_name="rec" model_name="rec"
yml_file="configs/rec/rec_mv3_none_bilstm_ctc.yml" yml_file="configs/rec/rec_mv3_none_bilstm_ctc.yml"
else else