change resolution and fix regression test for mmocr (open-mmlab#534)

lvhan028 · Jun 3, 2022 · ad08e07 · ad08e07
1 parent 6e4691a
commit ad08e07
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 8 deletions.
diff --git a/...ensorrt-fp16_dynamic-320x320-1344x2240.py → ...ensorrt-fp16_dynamic-320x320-2240x2240.py b/...ensorrt-fp16_dynamic-320x320-1344x2240.py → ...ensorrt-fp16_dynamic-320x320-2240x2240.py
@@ -9,5 +9,5 @@
  input=dict(
  min_shape=[1, 3, 320, 320],
  opt_shape=[1, 3, 600, 800],
- max_shape=[1, 3, 1344, 2240])))
+ max_shape=[1, 3, 2240, 2240])))
  ])
diff --git a/...ensorrt-int8_dynamic-320x320-1344x2240.py → ...ensorrt-int8_dynamic-320x320-2240x2240.py b/...ensorrt-int8_dynamic-320x320-1344x2240.py → ...ensorrt-int8_dynamic-320x320-2240x2240.py
@@ -9,5 +9,5 @@
  input=dict(
  min_shape=[1, 3, 320, 320],
  opt_shape=[1, 3, 600, 800],
- max_shape=[1, 3, 1344, 2240])))
+ max_shape=[1, 3, 2240, 2240])))
  ])
diff --git a/...ion_tensorrt_dynamic-320x320-1344x2240.py → ...ion_tensorrt_dynamic-320x320-2240x2240.py b/...ion_tensorrt_dynamic-320x320-1344x2240.py → ...ion_tensorrt_dynamic-320x320-2240x2240.py
@@ -7,5 +7,5 @@
  input=dict(
  min_shape=[1, 3, 320, 320],
  opt_shape=[1, 3, 600, 800],
- max_shape=[1, 3, 1344, 2240])))
+ max_shape=[1, 3, 2240, 2240])))
  ])
diff --git a/docs/zh_cn/04-developer-guide/do_regression_test.md b/docs/zh_cn/04-developer-guide/do_regression_test.md
@@ -180,7 +180,7 @@ tensorrt:
  convert_image: *convert_image_det
  backend_test: *default_backend_test
  sdk_config: *sdk_detection_dynamic
- deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-1024x1824.py
+ deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-2240x2240.py
 
 openvino:
  # 此处省略，内容同上
@@ -236,7 +236,7 @@ models:
 | 2 | crnn | ../mmocr/configs/textrecog/crnn/crnn_academic_dataset.py | Text Recognition | ${WORK_DIR}/mmocr/crnn/onnxruntime/static/crnn_academic-a723a1c5 | x| SDK-onnxruntime | configs/mmocr/text-recognition/text-recognition_sdk_dynamic.py | static | fp32 | True | - | x | False |
 | 3 | dbnet| ../mmocr/configs/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py | Text Detection | ../mmdeploy_checkpoints/mmocr/dbnet/dbnet_r18_fpnc_sbn_1200e_icdar2015_20210329-ba3ab597.pth | ICDAR2015 | Pytorch| -| - | - | - | 0.795 | - | -|
 | 4 | dbnet| ../mmocr/configs/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py | Text Detection | ../mmdeploy_checkpoints/mmocr/dbnet/dbnet_r18_fpnc_sbn_1200e_icdar2015_20210329-ba3ab597.pth | ICDAR | onnxruntime | configs/mmocr/text-detection/text-detection_onnxruntime_dynamic.py | dynamic | fp32 | True | - | - | True|
-| 5 | dbnet| ../mmocr/configs/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py | Text Detection | ${WORK_DIR}/mmocr/dbnet/tensorrt/dynamic/dbnet_r18_fpnc_sbn_1200e_icdar2015_20210329-ba3ab597/end2end.engine | ICDAR | tensorrt | configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-1024x1824.py | dynamic | fp16 | True | 0.793302 | - | True|
+| 5 | dbnet| ../mmocr/configs/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py | Text Detection | ${WORK_DIR}/mmocr/dbnet/tensorrt/dynamic/dbnet_r18_fpnc_sbn_1200e_icdar2015_20210329-ba3ab597/end2end.engine | ICDAR | tensorrt | configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-2240x2240.py | dynamic | fp16 | True | 0.793302 | - | True|
 | 6 | dbnet| ../mmocr/configs/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py | Text Detection | ${WORK_DIR}/mmocr/dbnet/tensorrt/dynamic/dbnet_r18_fpnc_sbn_1200e_icdar2015_20210329-ba3ab597 | ICDAR | SDK-tensorrt | configs/mmocr/text-detection/text-detection_sdk_dynamic.py | dynamic | fp16 | True | 0.795073 | - | True|
 
 

diff --git a/tests/regression/mmocr.yml b/tests/regression/mmocr.yml
@@ -75,19 +75,19 @@ tensorrt:
  convert_image: *convert_image_det
  backend_test: *default_backend_test
  sdk_config: *sdk_detection_dynamic
- deploy_config: configs/mmocr/text-detection/text-detection_tensorrt_dynamic-320x320-1024x1824.py
+ deploy_config: configs/mmocr/text-detection/text-detection_tensorrt_dynamic-320x320-2240x2240.py
 
  pipeline_trt_detection_dynamic_fp16: &pipeline_trt_detection_dynamic_fp16
  convert_image: *convert_image_det
  backend_test: *default_backend_test
  sdk_config: *sdk_detection_dynamic
- deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-1024x1824.py
+ deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-fp16_dynamic-320x320-2240x2240.py
 
  pipeline_trt_detection_dynamic_int8: &pipeline_trt_detection_dynamic_int8
  convert_image: *convert_image_det
  backend_test: *default_backend_test
  sdk_config: *sdk_detection_dynamic
- deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-int8_dynamic-320x320-1024x1824.py
+ deploy_config: configs/mmocr/text-detection/text-detection_tensorrt-int8_dynamic-320x320-2240x2240.py
 
  # ======= recognition =======
  pipeline_trt_recognition_static_fp32: &pipeline_trt_recognition_static_fp32