mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-24 23:33:08 +08:00
fix ch_ppocr_mobile_v2_0_det benchmark fps diff
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user