From de9b3d82d7ec211856ae3e5809754996b5ffe33d Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Fri, 4 Mar 2022 23:44:16 +0800 Subject: [PATCH] [TIPC] add inference benchmark param (#5301) --- .../cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- ...e_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt | 4 +++- .../cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- ...ascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt | 4 +++- test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt | 4 +++- .../dota/s2anet_alignconv_2x_spine_train_infer_python.txt | 4 +++- .../configs/dota/s2anet_conv_2x_spine_train_infer_python.txt | 4 +++- .../face_detection/blazeface_1000e_train_infer_python.txt | 4 +++- .../blazeface_fpn_ssh_1000e_train_infer_python.txt | 4 +++- .../faster_rcnn_r101_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r50_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r50_vd_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt | 4 +++- .../faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt | 4 +++- ...ster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt | 4 +++- .../fcos/fcos_dcn_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- .../configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- .../keypoint/dark_hrnet_w32_256x192_train_infer_python.txt | 4 +++- .../keypoint/dark_hrnet_w48_256x192_train_infer_python.txt | 4 +++- .../keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt | 4 +++- .../configs/keypoint/hrnet_w32_256x192_train_infer_python.txt | 4 +++- .../configs/keypoint/tinypose_128x96_train_infer_python.txt | 4 +++- .../mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt | 4 +++- .../mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt | 4 +++- .../mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt | 4 +++- .../mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- .../mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt | 4 +++- ...mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt | 4 +++- .../mot/fairmot_dla34_30e_1088x608_train_infer_python.txt | 4 +++- .../mot/fairmot_dla34_30e_576x320_train_infer_python.txt | 4 +++- ...rmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt | 4 +++- .../mot/jde_darknet53_30e_1088x608_train_infer_python.txt | 4 +++- .../mot/jde_darknet53_30e_576x320_train_infer_python.txt | 4 +++- .../configs/picodet/picodet_l_640_coco_train_infer_python.txt | 4 +++- .../picodet_lcnet_1_5x_416_coco_train_infer_python.txt | 4 +++- .../configs/picodet/picodet_m_416_coco_train_infer_python.txt | 4 +++- ...codet_mobilenetv3_large_1x_416_coco_train_infer_python.txt | 4 +++- .../picodet/picodet_r18_640_coco_train_infer_python.txt | 4 +++- .../configs/picodet/picodet_s_320_coco_train_infer_python.txt | 4 +++- .../picodet_shufflenetv2_1x_416_coco_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_mbv3_large_coco_FPGM_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_mbv3_large_coco_KL_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_mbv3_large_coco_PACT_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_mbv3_large_coco_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_mbv3_small_coco_train_infer_python.txt | 4 +++- .../configs/ppyolo/ppyolo_r18vd_coco_train_infer_python.txt | 4 +++- .../ppyolo_r50vd_dcn_1x_coco_FPGM_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_r50vd_dcn_1x_coco_KL_train_infer_python.txt | 4 +++- .../ppyolo_r50vd_dcn_1x_coco_PACT_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_r50vd_dcn_1x_coco_train_infer_python.txt | 4 +++- .../ppyolo/ppyolo_tiny_650e_coco_train_infer_python.txt | 4 +++- .../ppyolov2_r101vd_dcn_365e_coco_train_infer_python.txt | 4 +++- .../ppyolov2_r50vd_dcn_365e_coco_train_infer_python.txt | 4 +++- .../solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt | 4 +++- .../solov2/solov2_r50_enhance_coco_train_infer_python.txt | 4 +++- .../solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt | 4 +++- .../ssd/ssdlite_mobilenet_v1_300_coco_train_infer_python.txt | 4 +++- .../ttfnet/ttfnet_darknet53_1x_coco_train_infer_python.txt | 4 +++- .../yolov3_darknet53_270e_coco_FPGM_train_infer_python.txt | 4 +++- .../yolov3_darknet53_270e_coco_KL_train_infer_python.txt | 4 +++- .../yolov3_darknet53_270e_coco_PACT_train_infer_python.txt | 4 +++- .../yolov3/yolov3_darknet53_270e_coco_train_infer_python.txt | 4 +++- .../yolov3_mobilenet_v1_270e_coco_train_infer_python.txt | 4 +++- ...yolov3_mobilenet_v3_large_270e_coco_train_infer_python.txt | 4 +++- .../yolov3/yolov3_r34_270e_coco_train_infer_python.txt | 4 +++- .../yolov3/yolov3_r50vd_dcn_270e_coco_train_infer_python.txt | 4 +++- test_tipc/test_train_inference_python.sh | 4 ++-- 72 files changed, 215 insertions(+), 73 deletions(-) diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt index 0578a6d27..57f5eb868 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index 78de71fe9..10f0d36e3 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt index 8a35c39ef..8fcfce9c6 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index 77fe433d3..4bfdb61ac 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt index 6f1a19a97..87f386626 100644 --- a/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt +++ b/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/spine_coco/test/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,1024,1024]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt index 440b58675..fcbbb73df 100644 --- a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/spine_coco/test/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,1024,1024]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt index 886dbb423..aeabcd26e 100644 --- a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/spine_coco/test/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,1024,1024]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt b/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt index ff7790154..3d32e3642 100644 --- a/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt +++ b/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt index 6813acd10..bcb7263d0 100644 --- a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt +++ b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt index 3e60b3fdd..51944f3fa 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt index d350a3d4c..bda148953 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index 7493d7636..b235329e6 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt index c9d8e4893..5548ec911 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt index b02bfba7a..d3ae5f964 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt index feb237bd8..6cbc73d33 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 4af0fd24a..c4c5f6fe3 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 @@ -54,4 +54,6 @@ batch_size:2|8 fp_items:fp32|fp16 epoch: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,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt index 4565b4ad6..506d286a5 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index 62c543cc1..431a6d276 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt index 2739a6ab4..e49a639bf 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index dd008ccaf..2846b0367 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_infer_python.txt index 507808622..50f1dc8c6 100644 --- a/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 02fa59109..28eb4fe41 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 @@ -54,4 +54,6 @@ batch_size:2|8 fp_items:fp32|fp16 epoch: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,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt index f230eec87..fecb5cb7e 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/keypoint_infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,256,192]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt index 57f284a4f..f27836688 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/keypoint_infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,256,192]},{float32,[2]},{float32,[2]}] \ 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 574a6bd64..e9b77abe0 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 @@ -54,4 +54,6 @@ batch_size:20|24 fp_items:fp32|fp16 epoch: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,512,512]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 99b514390..21b9ecacc 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 @@ -54,4 +54,6 @@ batch_size:64|160 fp_items:fp32|fp16 epoch: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,256,192]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 cf12efceb..f56503065 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/keypoint_infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,128,96]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt index a614b23fc..e96e9d82b 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index d6b4574d7..168b5833e 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ 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 89ad118fc..3a6c19af1 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 @@ -54,4 +54,6 @@ batch_size:2|4 fp_items:fp32|fp16 epoch: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,800,1344]},{float32,[2]},{float32,[2]}] \ 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 d0ce936a4..bab73d982 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 @@ -54,4 +54,6 @@ batch_size:2|4 fp_items:fp32|fp16 epoch: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,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index 3dad7d4f1..0e933dc07 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index f408e73c7..8edc67e5c 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True ---trt_max_shape:1600 \ No newline at end of file +--trt_max_shape:1600 +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 2806d3c3d..6a7577fcf 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 @@ -54,4 +54,6 @@ batch_size:6|22 fp_items:fp32|fp16 epoch: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,608,1088]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt index 0db25dabc..f7e302cf1 100644 --- a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt +++ b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/pptracking/python/mot_jde_infer.py --video_file:./dataset/mot/test.mp4 --save_log_path:null --run_benchmark:False -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,576]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt index a6be7a9ac..b0a162494 100644 --- a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/pptracking/python/mot_jde_infer.py --video_file:./dataset/mot/test.mp4 --save_log_path:null --run_benchmark:False -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,576]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 e196d62cb..20081cd5c 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 @@ -54,4 +54,6 @@ batch_size:4|14 fp_items:fp32|fp16 epoch: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,608,1088]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt index b14e0ff94..8698d95de 100644 --- a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt +++ b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/pptracking/python/mot_jde_infer.py --video_file:./dataset/mot/test.mp4 --save_log_path:null --run_benchmark:False -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,576]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt index e8e79b357..528670fed 100644 --- a/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,640,640]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 cfe2dd7be..f3c4e32f8 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,416,416]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt index fce7000d0..6fe0967fa 100644 --- a/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,416,416]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt index 13e8b6336..eac66ffe6 100644 --- a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,416,416]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt index a5077a1ac..c8e1c449e 100644 --- a/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,640,640]},{float32,[2]},{float32,[2]}] \ 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 f918fda0c..8fefc47f1 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt index 66a394c13..e9e84268a 100644 --- a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,416,416]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_FPGM_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_FPGM_train_infer_python.txt index 670fe5f92..6da79c877 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_FPGM_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_FPGM_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_KL_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_KL_train_infer_python.txt index 34713c606..b4c148c9b 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_KL_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_KL_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_PACT_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_PACT_train_infer_python.txt index dfbd010e7..8cc78ffe3 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_PACT_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_PACT_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ 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 f69855fea..d8703b856 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_infer_python.txt index 705e12e22..2fdd6e731 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_infer_python.txt index 51f7856e0..79b1a23c1 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,512,512]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_FPGM_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_FPGM_train_infer_python.txt index f7dfd749d..459c5f5b5 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_FPGM_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_FPGM_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_KL_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_KL_train_infer_python.txt index 06f68578b..0839d59cf 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_KL_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_KL_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_PACT_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_PACT_train_infer_python.txt index eb62c13f7..e69b58535 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_PACT_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_PACT_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ 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 e13496072..027f8855d 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ 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 d672044e9..b72910f16 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,320,320]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_infer_python.txt b/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_infer_python.txt index 52b7ef256..aa865f400 100644 --- a/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_infer_python.txt +++ b/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,640,640]},{float32,[2]},{float32,[2]}] \ 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 d37913a6f..111b2b172 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 @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,640,640]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt index 32cf4cf64..f144ff2f7 100644 --- a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt +++ b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:False -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt index 98fd8db67..7efa37d04 100644 --- a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt +++ b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,512,864]},{float32,[2]},{float32,[2]}] \ 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 169eaa4bc..84198058b 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 @@ -54,4 +54,6 @@ batch_size:2|4 fp_items:fp32|fp16 epoch: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,800,1344]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_infer_python.txt b/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_infer_python.txt index 22fbafd05..3db17efff 100644 --- a/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_infer_python.txt +++ b/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,300,300]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_infer_python.txt b/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_infer_python.txt index e0fd1846f..6eeb5abb1 100644 --- a/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,512,512]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_FPGM_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_FPGM_train_infer_python.txt index 092a7d567..a552efd23 100644 --- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_FPGM_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_FPGM_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_KL_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_KL_train_infer_python.txt index 0c33ada32..db1517a77 100644 --- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_KL_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_KL_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_PACT_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_PACT_train_infer_python.txt index c4b6a109b..c405df9a6 100644 --- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_PACT_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_PACT_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file 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 7c5bff379..ac312961f 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 @@ -54,4 +54,6 @@ batch_size:8 fp_items:fp32|fp16 epoch: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,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_infer_python.txt index 843dd7cae..7db3f888b 100644 --- a/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_infer_python.txt index 0966741ee..6b4325330 100644 --- a/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_infer_python.txt index a2e7ed6a8..f199d0e1b 100644 --- a/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_infer_python.txt b/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_infer_python.txt index c105a8a13..d96fafb3f 100644 --- a/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_infer_python.txt +++ b/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_infer_python.txt @@ -48,4 +48,6 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params=========================== +random_infer_input:[{float32,[3,608,608]},{float32,[2]},{float32,[2]}] \ No newline at end of file diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index 18711603a..e76c60a53 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -263,7 +263,7 @@ else fi if [ ${autocast} = "amp" ]; then - set_autocast="--fp16" + set_autocast="--amp" else set_autocast=" " fi @@ -299,7 +299,7 @@ else if [ ${run_export} != "null" ]; then # run export model set_export_weight=$(func_set_params "${export_weight_key}" "${save_log}/${model_name}/${train_model_name}") - set_save_export_dir=$(func_set_params "${save_export_key}" "${save_export_value}") + set_save_export_dir=$(func_set_params "${save_export_key}" "${save_log}") export_cmd="${python} ${run_export} ${set_export_weight} ${set_filename} ${set_save_export_dir} " eval $export_cmd status_check $? "${export_cmd}" "${status_log}" -- GitLab