提交 9d450d8a 编写于 作者: 文幕地方's avatar 文幕地方

fix ch_ppocr_mobile_v2_0_det benchmark fps diff

上级 d76b7f8b
......@@ -13,7 +13,7 @@ train_infer_img_dir:./train_data/icdar2015/text_localization/ch4_test_images/
null:null
##
trainer:norm_train
norm_train:tools/train.py -c configs/det/ch_ppocr_v2.0/ch_det_mv3_db_v2.0.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained Global.print_batch_step=1 Train.loader.shuffle=false
norm_train:tools/train.py -c configs/det/ch_ppocr_v2.0/ch_det_mv3_db_v2.0.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained Global.print_batch_step=2 Train.loader.shuffle=false
pact_train:null
fpgm_train:null
distill_train:null
......
......@@ -32,6 +32,13 @@ if [ ${MODE} = "benchmark_train" ];then
cd ./train_data/ && tar xf icdar2015_benckmark.tar
ln -s ./icdar2015_benckmark ./icdar2015
cd ../
if [[ ${model_name} =~ "ch_ppocr_mobile_v2_0_det" ]];then
# expand gt.txt 2 times
cd ./train_data/icdar2015/text_localization
for i in `seq 2`;do cp train_icdar2015_label.txt dup$i.txt;done
cat dup* > train_icdar2015_label.txt && rm -rf dup*
cd ../../../
fi
fi
if [[ ${model_name} =~ "ch_ppocr_server_v2_0_det" || ${model_name} =~ "ch_PP-OCRv3_det" ]];then
rm -rf ./train_data/icdar2015
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册