mmyolo/configs/deploy/detection_tensorrt-fp16_static-640x640.py at main · open-mmlab/mmyolo

1

2

3

4

5

6

7

8

9

10

11

12

13

14

_base_ = ['./base_static.py']

onnx_config = dict(input_shape=(640, 640))

backend_config = dict(

type='tensorrt',

common_config=dict(fp16_mode=True, max_workspace_size=1 << 30),

model_inputs=[

dict(

input_shapes=dict(

input=dict(

min_shape=[1, 3, 640, 640],

opt_shape=[1, 3, 640, 640],

max_shape=[1, 3, 640, 640])))

])

use_efficientnms = False # whether to replace TRTBatchedNMS plugin with EfficientNMS plugin # noqa E501

Provide feedback