[Fix] fix ssd ncnn ut (#692)

hanrui1sensetime · web-flow · commit affc5ae3ff56 · 2022-07-04T15:17:30.000+08:00
* fix ssd ncnn ut

* fix yapf
diff --git a/tests/test_codebase/test_mmdet/test_mmdet_models.py b/tests/test_codebase/test_mmdet/test_mmdet_models.py
@@ -1409,18 +1409,19 @@ def test_ssd_head_get_bboxes__ncnn(is_dynamic: bool):
         'img_shape': (s, s, 3)
     }]
     output_names = ['output']
-    input_names = ['input']
+    input_names = []
+    for i in range(6):
+        input_names.append('cls_scores_' + str(i))
+        input_names.append('bbox_preds_' + str(i))
     dynamic_axes = None
     if is_dynamic:
         dynamic_axes = {
-            input_names[0]: {
-                2: 'height',
-                3: 'width'
-            },
             output_names[0]: {
                 1: 'num_dets',
             }
         }
+        for input_name in input_names:
+            dynamic_axes[input_name] = {2: 'height', 3: 'width'}
     deploy_cfg = mmcv.Config(
         dict(
             backend_config=dict(type=Backend.NCNN.value),
