Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
81bd4104
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
81bd4104
编写于
5月 26, 2022
作者:
A
andyjpaddle
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update train infer python for tipc
上级
e2fafc30
变更
39
展开全部
显示空白变更内容
内联
并排
Showing
39 changed file
with
226 addition
and
177 deletion
+226
-177
test_tipc/configs/ch_PP-OCRv2_det/train_infer_python.txt
test_tipc/configs/ch_PP-OCRv2_det/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv2_det_PACT/train_infer_python.txt
..._tipc/configs/ch_PP-OCRv2_det_PACT/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt
..._tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv3_det/train_infer_python.txt
test_tipc/configs/ch_PP-OCRv3_det/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv3_det_PACT/train_infer_python.txt
..._tipc/configs/ch_PP-OCRv3_det_PACT/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv3_rec/train_infer_python.txt
test_tipc/configs/ch_PP-OCRv3_rec/train_infer_python.txt
+5
-5
test_tipc/configs/ch_PP-OCRv3_rec_PACT/train_infer_python.txt
..._tipc/configs/ch_PP-OCRv3_rec_PACT/train_infer_python.txt
+5
-5
test_tipc/configs/ch_ppocr_mobile_v2.0_det/train_infer_python.txt
...c/configs/ch_ppocr_mobile_v2.0_det/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_mobile_v2.0_det_FPGM/train_infer_python.txt
...figs/ch_ppocr_mobile_v2.0_det_FPGM/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_mobile_v2.0_det_PACT/train_infer_python.txt
...figs/ch_ppocr_mobile_v2.0_det_PACT/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt
...c/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
...figs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
...figs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_server_v2.0_det/train_infer_python.txt
...c/configs/ch_ppocr_server_v2.0_det/train_infer_python.txt
+4
-4
test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt
...c/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt
+3
-3
test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt
test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt
+3
-3
test_tipc/configs/det_mv3_east_v2.0/train_infer_python.txt
test_tipc/configs/det_mv3_east_v2.0/train_infer_python.txt
+2
-2
test_tipc/configs/det_mv3_pse_v2.0/train_infer_python.txt
test_tipc/configs/det_mv3_pse_v2.0/train_infer_python.txt
+2
-2
test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt
test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt
+3
-3
test_tipc/configs/det_r50_db_v2.0/train_infer_python.txt
test_tipc/configs/det_r50_db_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
..._tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
+2
-2
test_tipc/configs/det_r50_vd_pse_v2_0/train_infer_python.txt
test_tipc/configs/det_r50_vd_pse_v2_0/train_infer_python.txt
+2
-2
test_tipc/configs/det_r50_vd_sast_icdar15_v2.0/train_infer_python.txt
...nfigs/det_r50_vd_sast_icdar15_v2.0/train_infer_python.txt
+2
-2
test_tipc/configs/det_r50_vd_sast_totaltext_v2.0/train_infer_python.txt
...igs/det_r50_vd_sast_totaltext_v2.0/train_infer_python.txt
+2
-2
test_tipc/configs/en_server_pgnetA/train_infer_python.txt
test_tipc/configs/en_server_pgnetA/train_infer_python.txt
+2
-2
test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt
test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt
+3
-3
test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt
...nfigs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt
...configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt
...onfigs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt
...onfigs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_r31_sar/train_infer_python.txt
test_tipc/configs/rec_r31_sar/train_infer_python.txt
+4
-4
test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt
...gs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt
...figs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt
...igs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt
...igs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt
+3
-3
test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt
...pc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt
+3
-3
test_tipc/prepare.sh
test_tipc/prepare.sh
+95
-46
test_tipc/test_train_inference_python.sh
test_tipc/test_train_inference_python.sh
+3
-3
未找到文件。
test_tipc/configs/ch_PP-OCRv2_det/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv2_det
model_name:ch_PP
-
OCRv2_det
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_PP-OCRv2_det_PACT/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv2_det_PACT
model_name:ch_PP
-
OCRv2_det_PACT
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=1|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:
PPOCRv2_ocr
_rec
model_name:
ch_PP-OCRv2
_rec
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=128
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_rec.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv2_rec_PACT
model_name:ch_PP
-
OCRv2_rec_PACT
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=6|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=6|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=128
Global.pretrained_model:pretrain_models/ch_PP-OCRv2_rec_train/best_accuracy
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:True
inference:tools/infer/predict_rec.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_PP-OCRv3_det/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv3_det
model_name:ch_PP
-
OCRv3_det
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_PP-OCRv3_det_PACT/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv3_det_PACT
model_name:ch_PP
-
OCRv3_det_PACT
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=1|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_PP-OCRv3_rec/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:
PPOCRv3_ocr
_rec
model_name:
ch_PP-OCRv3
_rec
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=128
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_PP-OCRv3_rec_PACT/train_infer_python.txt
浏览文件 @
81bd4104
===========================train_params===========================
model_name:ch_PPOCRv3_rec_PACT
model_name:ch_PP
-
OCRv3_rec_PACT
python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:fp32
Global.epoch_num:lite_train_lite_infer=6|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=6|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=128
Global.pretrained_model:pretrain_models/ch_PP-OCRv3_rec_train/best_accuracy
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:True
inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_det/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=100|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=100|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c configs/det/ch_ppocr_v2.0/ch_det_mv3_db_v2
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_det_FPGM/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=5|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=5|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_det_PACT/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=20|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=20|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=128|whole_train_whole_infer=128
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c configs/rec/rec_icdar15_train.yml -o
infer_quant:False
inference:tools/infer/predict_rec.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=128|whole_train_whole_infer=128
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_rec.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=128|whole_train_whole_infer=128
Global.checkpoints:null
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ppocr_keys_v1.txt --rec_image_shape="3,32,100"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/ch_ppocr_server_v2.0_det/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -4,7 +4,7 @@ python:python3.7
gpu_list:0|0,1
Global.use_gpu:True|True
Global.auto_cast:null
Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=
30
0
Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=
5
0
Global.save_model_dir:./output/
Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_lite_infer=4
Global.pretrained_model:null
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c configs/det/ch_ppocr_v2.0/ch_det_res18_db_
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/ch_ppocr_server_v2.0_rec
infer_quant:False
inference:tools/infer/predict_rec.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c configs/det/det_mv3_db.yml -o
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/det_mv3_east_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_mv3_east_v2.0/det_mv
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
...
...
test_tipc/configs/det_mv3_pse_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_mv3_pse_v2.0/det_mv3
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
...
...
test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:null
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/det_r50_db_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c configs/det/det_r50_vd_db.yml -o
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
|True
--use_tensorrt:False
--precision:fp32|fp16|int8
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
...
...
test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_east_v2_0/det
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
...
...
test_tipc/configs/det_r50_vd_pse_v2_0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_pse_v2_0/det_
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
...
...
test_tipc/configs/det_r50_vd_sast_icdar15_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_sast_icdar15_
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|int8
...
...
test_tipc/configs/det_r50_vd_sast_totaltext_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_sast_totaltex
infer_quant:False
inference:tools/infer/predict_det.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|int8
...
...
test_tipc/configs/en_server_pgnetA/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,8 +39,8 @@ infer_export:tools/export_model.py -c configs/e2e/e2e_r50_vd_pg.yml -o
infer_quant:False
inference:tools/infer/predict_e2e.py
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
...
...
test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_mtb_nrtr/rec_mtb_nrt
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/EN_symbol_dict.txt --rec_image_shape="1,32,100" --rec_algorithm="NRTR"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_mv3_none_bilstm_ctc_
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_mv3_none_none_ctc_v2
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_mv3_tps_bilstm_att_v
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" --rec_algorithm="RARE" --min_subgraph_size=5
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_mv3_tps_bilstm_ctc_v
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" --rec_algorithm="StarNet"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r31_sar/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -38,11 +38,11 @@ train_model:./inference/rec_r31_sar_train/best_accuracy
infer_export:tools/export_model.py -c test_tipc/configs/rec_r31_sar/rec_r31_sar.yml -o
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.txt --rec_image_shape="3,48,48,160" --rec_algorithm="SAR"
--use_gpu:True
|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--use_gpu:True
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_r34_vd_none_bilstm_c
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_r34_vd_none_none_ctc
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_r34_vd_tps_bilstm_at
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" --rec_algorithm="RARE" --min_subgraph_size=5
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_r34_vd_tps_bilstm_ct
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" --rec_algorithm="StarNet"
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt
浏览文件 @
81bd4104
...
...
@@ -39,10 +39,10 @@ infer_export:tools/export_model.py -c test_tipc/configs/rec_r50_fpn_vd_none_srn/
infer_quant:False
inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="1,64,256" --rec_algorithm="SRN" --use_space_char=False --min_subgraph_size=3
--use_gpu:True|False
--enable_mkldnn:
True|
False
--cpu_threads:
1|
6
--enable_mkldnn:False
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:
True|
False
--use_tensorrt:False
--precision:fp32|int8
--rec_model_dir:
--image_dir:./inference/rec_inference
...
...
test_tipc/prepare.sh
浏览文件 @
81bd4104
此差异已折叠。
点击以展开。
test_tipc/test_train_inference_python.sh
浏览文件 @
81bd4104
...
...
@@ -142,9 +142,9 @@ function func_inference(){
for
use_gpu
in
${
use_gpu_list
[*]
}
;
do
if
[
${
use_gpu
}
=
"False"
]
||
[
${
use_gpu
}
=
"cpu"
]
;
then
for
use_mkldnn
in
${
use_mkldnn_list
[*]
}
;
do
if
[
${
use_mkldnn
}
=
"False"
]
&&
[
${
_flag_quant
}
=
"True"
]
;
then
continue
fi
#
if [ ${use_mkldnn} = "False" ] && [ ${_flag_quant} = "True" ]; then
#
continue
#
fi
for
threads
in
${
cpu_threads_list
[*]
}
;
do
for
batch_size
in
${
batch_size_list
[*]
}
;
do
for
precision
in
${
precision_list
[*]
}
;
do
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录