From cdabce67bdd79d61679659f480a26e96891b426b Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 10 Feb 2022 10:07:02 +0800
Subject: [PATCH] add east

---
 .../det_r50_vd_east.yml                             |  0
 .../train_infer_python.txt                          | 13 +++++++++----
 test_tipc/prepare.sh                                |  2 +-
 3 files changed, 10 insertions(+), 5 deletions(-)
 rename test_tipc/configs/{det_r50_vd_east_v2.0 => det_r50_vd_east_v2_0}/det_r50_vd_east.yml (100%)
 rename test_tipc/configs/{det_r50_vd_east_v2.0 => det_r50_vd_east_v2_0}/train_infer_python.txt (80%)

diff --git a/test_tipc/configs/det_r50_vd_east_v2.0/det_r50_vd_east.yml b/test_tipc/configs/det_r50_vd_east_v2_0/det_r50_vd_east.yml
similarity index 100%
rename from test_tipc/configs/det_r50_vd_east_v2.0/det_r50_vd_east.yml
rename to test_tipc/configs/det_r50_vd_east_v2_0/det_r50_vd_east.yml
diff --git a/test_tipc/configs/det_r50_vd_east_v2.0/train_infer_python.txt b/test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
similarity index 80%
rename from test_tipc/configs/det_r50_vd_east_v2.0/train_infer_python.txt
rename to test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
index c7c8d089..bc149ac8 100644
--- a/test_tipc/configs/det_r50_vd_east_v2.0/train_infer_python.txt
+++ b/test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
@@ -1,5 +1,5 @@
 ===========================train_params===========================
-model_name:det_r50_vd_east_v2.0
+model_name:det_r50_vd_east_v2_0
 python:python3.7
 gpu_list:0
 Global.use_gpu:True|True
@@ -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 test_tipc/configs/det_r50_vd_east_v2.0/det_r50_vd_east.yml -o 
+norm_train:tools/train.py -c test_tipc/configs/det_r50_vd_east_v2_0/det_r50_vd_east.yml -o 
 pact_train:null
 fpgm_train:null
 distill_train:null
@@ -27,7 +27,7 @@ null:null
 ===========================infer_params===========================
 Global.save_inference_dir:./output/
 Global.checkpoints:
-norm_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_east_v2.0/det_r50_vd_east.yml -o 
+norm_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_east_v2_0/det_r50_vd_east.yml -o 
 quant_export:null 
 fpgm_export:null
 distill_export:null
@@ -35,7 +35,7 @@ export1:null
 export2:null
 ##
 train_model:./inference/det_r50_vd_east_v2.0_train/best_accuracy
-infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_east_v2.0/det_r50_vd_east.yml  -o
+infer_export:tools/export_model.py -c test_tipc/configs/det_r50_vd_east_v2_0/det_r50_vd_east.yml  -o
 infer_quant:False
 inference:tools/infer/predict_det.py
 --use_gpu:True|False
@@ -49,3 +49,8 @@ inference:tools/infer/predict_det.py
 --save_log_path:null
 --benchmark:True
 --det_algorithm:EAST
+===========================train_benchmark_params==========================
+batch_size:8
+fp_items:fp32|fp16
+epoch:2
+--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
\ No newline at end of file
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index ea114697..c78c321c 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -231,7 +231,7 @@ elif [ ${MODE} = "whole_infer" ];then
         wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_east_v2.0_train.tar --no-check-certificate
         cd ./inference/ && tar xf det_mv3_east_v2.0_train.tar & cd ../
     fi
-    if [ ${model_name} == "det_r50_vd_east_v2.0" ]; then
+    if [ ${model_name} == "det_r50_vd_east_v2_0" ]; then
         wget -nc -P ./inference/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_r50_vd_east_v2.0_train.tar --no-check-certificate
         cd ./inference/ && tar xf det_r50_vd_east_v2.0_train.tar & cd ../
     fi
-- 
GitLab