fix telescope tipc error
parent
210135cc94
commit
d1f57063fb
|
@ -51,7 +51,7 @@ Metric:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSetSR
|
name: LMDBDataSetSR
|
||||||
data_dir: ./train_data/TextZoom/train
|
data_dir: ./train_data/TextZoom/test
|
||||||
transforms:
|
transforms:
|
||||||
- SRResize:
|
- SRResize:
|
||||||
imgH: 32
|
imgH: 32
|
||||||
|
|
|
@ -4,12 +4,12 @@ python:python3.7
|
||||||
gpu_list:0|0,1
|
gpu_list:0|0,1
|
||||||
Global.use_gpu:True|True
|
Global.use_gpu:True|True
|
||||||
Global.auto_cast:null
|
Global.auto_cast:null
|
||||||
Global.epoch_num:lite_train_lite_infer=2|whole_train_whole_infer=300
|
Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=300
|
||||||
Global.save_model_dir:./output/
|
Global.save_model_dir:./output/
|
||||||
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=16
|
Train.loader.batch_size_per_card:lite_train_lite_infer=16|whole_train_whole_infer=16
|
||||||
Global.pretrained_model:null
|
Global.pretrained_model:null
|
||||||
train_model_name:latest
|
train_model_name:latest
|
||||||
train_infer_img_dir:./inference/sr_inference
|
train_infer_img_dir:./inference/rec_inference
|
||||||
null:null
|
null:null
|
||||||
##
|
##
|
||||||
trainer:norm_train
|
trainer:norm_train
|
||||||
|
@ -21,7 +21,7 @@ null:null
|
||||||
null:null
|
null:null
|
||||||
##
|
##
|
||||||
===========================eval_params===========================
|
===========================eval_params===========================
|
||||||
eval:tools/eval.py -c test_tipc/configs/sr_telescope/sr_telescope.yml -o
|
eval:null
|
||||||
null:null
|
null:null
|
||||||
##
|
##
|
||||||
===========================infer_params===========================
|
===========================infer_params===========================
|
||||||
|
@ -44,8 +44,8 @@ inference:tools/infer/predict_sr.py --sr_image_shape="1,32,128" --rec_algorithm=
|
||||||
--rec_batch_num:1
|
--rec_batch_num:1
|
||||||
--use_tensorrt:False
|
--use_tensorrt:False
|
||||||
--precision:fp32
|
--precision:fp32
|
||||||
--rec_model_dir:
|
--sr_model_dir:
|
||||||
--image_dir:./inference/sr_inference
|
--image_dir:./inference/rec_inference
|
||||||
--save_log_path:./test/output/
|
--save_log_path:./test/output/
|
||||||
--benchmark:True
|
--benchmark:True
|
||||||
null:null
|
null:null
|
||||||
|
|
|
@ -150,6 +150,7 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
|
||||||
# pretrain lite train data
|
# pretrain lite train data
|
||||||
wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
|
wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
|
||||||
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_db_v2.0_train.tar --no-check-certificate
|
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_db_v2.0_train.tar --no-check-certificate
|
||||||
|
cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../
|
||||||
if [[ ${model_name} =~ "ch_PP-OCRv2_det" ]];then
|
if [[ ${model_name} =~ "ch_PP-OCRv2_det" ]];then
|
||||||
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_distill_train.tar --no-check-certificate
|
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_distill_train.tar --no-check-certificate
|
||||||
cd ./pretrain_models/ && tar xf ch_PP-OCRv2_det_distill_train.tar && cd ../
|
cd ./pretrain_models/ && tar xf ch_PP-OCRv2_det_distill_train.tar && cd ../
|
||||||
|
@ -179,7 +180,6 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
|
||||||
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/ppstructure/models/tablemaster/table_structure_tablemaster_train.tar --no-check-certificate
|
wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/ppstructure/models/tablemaster/table_structure_tablemaster_train.tar --no-check-certificate
|
||||||
cd ./pretrain_models/ && tar xf table_structure_tablemaster_train.tar && cd ../
|
cd ./pretrain_models/ && tar xf table_structure_tablemaster_train.tar && cd ../
|
||||||
fi
|
fi
|
||||||
cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../
|
|
||||||
rm -rf ./train_data/icdar2015
|
rm -rf ./train_data/icdar2015
|
||||||
rm -rf ./train_data/ic15_data
|
rm -rf ./train_data/ic15_data
|
||||||
rm -rf ./train_data/pubtabnet
|
rm -rf ./train_data/pubtabnet
|
||||||
|
|
Loading…
Reference in New Issue