From 9d450d8a2be9579e0b151210207e95abb9971edc Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Fri, 2 Sep 2022 13:51:37 +0000
Subject: [PATCH] fix ch_ppocr_mobile_v2_0_det benchmark fps diff

---
 .../ch_ppocr_mobile_v2_0_det/train_infer_python.txt        | 2 +-
 test_tipc/prepare.sh                                       | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/test_tipc/configs/ch_ppocr_mobile_v2_0_det/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2_0_det/train_infer_python.txt
index f3aa9d0f..4112e649 100644
--- a/test_tipc/configs/ch_ppocr_mobile_v2_0_det/train_infer_python.txt
+++ b/test_tipc/configs/ch_ppocr_mobile_v2_0_det/train_infer_python.txt
@@ -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
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index f1badbd8..49b514b3 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -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
-- 
GitLab