[Fix] fix ssd ncnn ut (#692)

* fix ssd ncnn ut

* fix yapf
pull/700/head
hanrui1sensetime 2022-07-04 15:17:30 +08:00 committed by GitHub
parent 17a7d60c7d
commit affc5ae3ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions

View File

@ -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),