From f7890b1b9c85cbf5a749870a4c484bce3c946245 Mon Sep 17 00:00:00 2001
From: shangliang Xu <ghostxsl@users.noreply.github.com>
Date: Tue, 2 Aug 2022 19:48:36 +0800
Subject: [PATCH] [TIPC] fix pose benchmark shell and txt (#6564)

---
 ppdet/data/source/dataset.py                  |  6 ++--
 test_tipc/benchmark_train.sh                  |  1 +
 ...cnn_r50_fpn_1x_coco_train_infer_python.txt |  2 +-
 ...cos_r50_fpn_1x_coco_train_infer_python.txt |  2 +-
 ...gfl_r50_fpn_1x_coco_train_infer_python.txt |  2 +-
 ...hrnet_hrnet_w32_512_train_infer_python.txt |  2 +-
 .../hrnet_w32_256x192_train_infer_python.txt  |  2 +-
 .../tinypose_128x96_train_infer_python.txt    |  8 ++---
 ...sk_rcnn_r50_1x_coco_train_infer_python.txt |  7 +++++
 ...cnn_r50_fpn_1x_coco_train_infer_python.txt |  2 +-
 ..._dla34_30e_1088x608_train_infer_python.txt |  4 +--
 ...knet53_30e_1088x608_train_infer_python.txt |  2 +-
 ...lcnet_1_5x_416_coco_train_infer_python.txt |  8 ++---
 ...et_s_320_coco_lcnet_train_infer_python.txt | 12 +++----
 .../picodet_s_320_coco_train_infer_python.txt |  8 ++---
 ...olo_mbv3_large_coco_train_infer_python.txt |  8 ++---
 ...o_r50vd_dcn_1x_coco_train_infer_python.txt |  8 ++---
 ...yolo_tiny_650e_coco_train_infer_python.txt |  8 ++---
 ...r50vd_dcn_365e_coco_train_infer_python.txt |  8 ++---
 ...loe_crn_s_300e_coco_train_infer_python.txt |  8 ++---
 ...ov2_r50_fpn_1x_coco_train_infer_python.txt |  2 +-
 ...darknet53_270e_coco_train_infer_python.txt |  2 +-
 test_tipc/prepare.sh                          | 31 ++++++++++++-------
 23 files changed, 78 insertions(+), 65 deletions(-)

diff --git a/ppdet/data/source/dataset.py b/ppdet/data/source/dataset.py
index d8361eea9..d735cfc4a 100644
--- a/ppdet/data/source/dataset.py
+++ b/ppdet/data/source/dataset.py
@@ -68,20 +68,18 @@ class DetDataset(Dataset):
         return self
 
     def __getitem__(self, idx):
+        n = len(self.roidbs)
         if self.repeat > 1:
-            idx %= self.repeat
+            idx %= n
         # data batch
         roidb = copy.deepcopy(self.roidbs[idx])
         if self.mixup_epoch == 0 or self._epoch < self.mixup_epoch:
-            n = len(self.roidbs)
             idx = np.random.randint(n)
             roidb = [roidb, copy.deepcopy(self.roidbs[idx])]
         elif self.cutmix_epoch == 0 or self._epoch < self.cutmix_epoch:
-            n = len(self.roidbs)
             idx = np.random.randint(n)
             roidb = [roidb, copy.deepcopy(self.roidbs[idx])]
         elif self.mosaic_epoch == 0 or self._epoch < self.mosaic_epoch:
-            n = len(self.roidbs)
             roidb = [roidb, ] + [
                 copy.deepcopy(self.roidbs[np.random.randint(n)])
                 for _ in range(4)
diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh
index dd7471d03..0c3f56bd5 100644
--- a/test_tipc/benchmark_train.sh
+++ b/test_tipc/benchmark_train.sh
@@ -166,6 +166,7 @@ else
 fi
 
 if [[ ${model_name} =~ "higherhrnet" ]] || [[ ${model_name} =~ "hrnet" ]] || [[ ${model_name} =~ "tinypose" ]];then
+    echo "${model_name} run on full coco dataset"
     epoch=$(set_dynamic_epoch $device_num $epoch)
 else
     epoch=1
diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt
index 7bfad1c68..bb15bdd8e 100644
--- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt
@@ -53,7 +53,7 @@ inference:./deploy/python/infer.py
 batch_size:2|8
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:3
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt
index 136d1c30e..424269648 100644
--- a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt
@@ -53,7 +53,7 @@ inference:./deploy/python/infer.py
 batch_size:2|8
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:3
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_infer_python.txt
index 23722615b..ed377f90a 100644
--- a/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_infer_python.txt
@@ -53,6 +53,6 @@ inference:./deploy/python/infer.py
 batch_size:2|8
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:3
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
\ No newline at end of file
diff --git a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt
index b827e92eb..a652e7f4a 100644
--- a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt
+++ b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt
@@ -52,7 +52,7 @@ inference:./deploy/python/keypoint_infer.py
 ===========================train_benchmark_params==========================
 batch_size:20|24
 fp_items:fp32|fp16
-epoch:20
+epoch:1
 repeat:1
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
diff --git a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt
index 036c542a9..ce486db0e 100644
--- a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt
+++ b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt
@@ -52,7 +52,7 @@ null:null
 ===========================train_benchmark_params==========================
 batch_size:64|160
 fp_items:fp32|fp16
-epoch:40
+epoch:1
 repeat:1
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
diff --git a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt
index 738a34885..f6ca2c397 100644
--- a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt
+++ b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/keypoint_infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-random_infer_input:[{float32,[3,128,96]}]
 ===========================train_benchmark_params==========================
 batch_size:512
 fp_items:fp32|fp16
-epoch:25
+epoch:1
 repeat:1
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+random_infer_input:[{float32,[3,128,96]}]
\ No newline at end of file
diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt
index 0437aa929..957d44268 100644
--- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt
@@ -49,5 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 --trt_max_shape:1600
+===========================train_benchmark_params==========================
+batch_size:2|4
+fp_items:fp32|fp16
+epoch:1
+repeat:2
+--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
+flags:null
 ===========================infer_benchmark_params===========================
 numpy_infer_input:3x800x1344.npy
\ No newline at end of file
diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt
index f03b573c0..db6d2b00a 100644
--- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt
@@ -53,7 +53,7 @@ inference:./deploy/python/infer.py
 batch_size:2|4
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:2
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_infer_python.txt b/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_infer_python.txt
index 0cf5b61b8..5bc67a522 100644
--- a/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_infer_python.txt
+++ b/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_infer_python.txt
@@ -50,10 +50,10 @@ inference:./deploy/pptracking/python/mot_jde_infer.py
 --run_benchmark:False
 --trt_max_shape:1600
 ===========================train_benchmark_params==========================
-batch_size:6|22
+batch_size:6|20
 fp_items:fp32|fp16
 epoch:1
-repeat:2
+repeat:1
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_infer_python.txt b/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_infer_python.txt
index e5760fbea..519722a53 100644
--- a/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_infer_python.txt
+++ b/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_infer_python.txt
@@ -50,7 +50,7 @@ inference:./deploy/pptracking/python/mot_jde_infer.py
 --run_benchmark:False
 --trt_max_shape:1600
 ===========================train_benchmark_params==========================
-batch_size:4|14
+batch_size:4|12
 fp_items:fp32|fp16
 epoch:1
 repeat:1
diff --git a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt
index b5b0676d4..694fa8818 100644
--- a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt
+++ b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x416x416_2.npy
 ===========================train_benchmark_params==========================
 batch_size:80
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:30
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x416x416_2.npy
\ No newline at end of file
diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt
index 12874c646..f51595a9e 100644
--- a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt
+++ b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt
@@ -6,7 +6,7 @@ use_gpu:True
 auto_cast:null
 epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=300
 save_dir:null
-TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=128
+TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=64
 pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_s_320_coco_lcnet.pdparams
 trained_model_name:model_final.pdparams
 train_infer_img_dir:./dataset/coco/test2017/
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x320x320_2.npy
 ===========================train_benchmark_params==========================
-batch_size:128
+batch_size:64
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:25
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x320x320_2.npy
\ No newline at end of file
diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt
index d04567e1e..0613fe1f0 100644
--- a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt
+++ b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x320x320_2.npy
 ===========================train_benchmark_params==========================
 batch_size:128
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:50
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x320x320_2.npy
\ No newline at end of file
diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_infer_python.txt
index 1735bfe5a..87ddbdc35 100644
--- a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_infer_python.txt
+++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x320x320.npy
 ===========================train_benchmark_params==========================
 batch_size:24
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:10
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x320x320.npy
\ No newline at end of file
diff --git a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_infer_python.txt
index 8ab40462d..67235a343 100644
--- a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x608x608.npy
 ===========================train_benchmark_params==========================
 batch_size:24
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:10
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x608x608.npy
\ No newline at end of file
diff --git a/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_infer_python.txt
index 2fed9d1ba..f26d63aba 100644
--- a/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_infer_python.txt
+++ b/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x320x320.npy
 ===========================train_benchmark_params==========================
 batch_size:32
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:12
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x320x320.npy
\ No newline at end of file
diff --git a/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_infer_python.txt
index af0bb920f..89b0e6e2b 100644
--- a/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_infer_python.txt
+++ b/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 null:null
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x640x640.npy
 ===========================train_benchmark_params==========================
 batch_size:12
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:5
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x640x640.npy
\ No newline at end of file
diff --git a/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_infer_python.txt b/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_infer_python.txt
index 536d969b3..617ac7038 100644
--- a/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_infer_python.txt
+++ b/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_infer_python.txt
@@ -49,12 +49,12 @@ inference:./deploy/python/infer.py
 --save_log_path:null
 --run_benchmark:False
 --trt_max_shape:1600
-===========================infer_benchmark_params===========================
-numpy_infer_input:3x640x640.npy
 ===========================train_benchmark_params==========================
 batch_size:32
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:12
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
\ No newline at end of file
+flags:null
+===========================infer_benchmark_params===========================
+numpy_infer_input:3x640x640.npy
\ No newline at end of file
diff --git a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt
index 4cc4de189..8d81e6f5a 100644
--- a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt
+++ b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt
@@ -53,7 +53,7 @@ inference:./deploy/python/infer.py
 batch_size:2|4
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:2
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt
index 43e8e643c..ce2476249 100644
--- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt
+++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt
@@ -53,7 +53,7 @@ null:null
 batch_size:8
 fp_items:fp32|fp16
 epoch:1
-repeat:1
+repeat:3
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params===========================
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 252b59aa2..adf8d2dba 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -86,18 +86,25 @@ elif [ ${MODE} = "benchmark_train" ];then
     pip install -U pip
     pip install Cython
     pip install -r requirements.txt
-    # prepare lite benchmark coco data
-    wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/coco_benchmark.tar --no-check-certificate
-    cd ./dataset/coco/ && tar -xvf coco_benchmark.tar 
-    mv -u coco_benchmark/* ./
-    ls ./
-    cd ../../
-    # prepare lite benchmark mot data
-    wget -nc -P ./dataset/mot/ https://paddledet.bj.bcebos.com/data/mot_benchmark.tar --no-check-certificate
-    cd ./dataset/mot/ && tar -xvf mot_benchmark.tar
-    mv -u mot_benchmark/* ./
-    ls ./
-    cd ../../
+    if [[ ${model_name} =~ "higherhrnet" ]] || [[ ${model_name} =~ "hrnet" ]] || [[ ${model_name} =~ "tinypose" ]];then
+        wget -nc -P ./dataset/ https://bj.bcebos.com/v1/paddledet/data/coco.tar --no-check-certificate
+        cd ./dataset/ && tar -xf coco.tar
+        ls ./coco/
+        cd ../
+    else
+        # prepare lite benchmark coco data
+        wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/coco_benchmark.tar --no-check-certificate
+        cd ./dataset/coco/ && tar -xf coco_benchmark.tar
+        mv -u coco_benchmark/* ./
+        ls ./
+        cd ../../
+        # prepare lite benchmark mot data
+        wget -nc -P ./dataset/mot/ https://paddledet.bj.bcebos.com/data/mot_benchmark.tar --no-check-certificate
+        cd ./dataset/mot/ && tar -xf mot_benchmark.tar
+        mv -u mot_benchmark/* ./
+        ls ./
+        cd ../../
+    fi
 elif [ ${MODE} = "paddle2onnx_infer" ];then
     # install paddle2onnx
     ${python} -m pip install paddle2onnx
-- 
GitLab