| _base_ = ['./base_dynamic.py'] | |
| backend_config = dict( | |
| type='tensorrt', | |
| common_config=dict(fp16_mode=True, max_workspace_size=1 << 32), | |
| model_inputs=[ | |
| dict( | |
| input_shapes=dict( | |
| input=dict( | |
| min_shape=[1, 3, 64, 64], | |
| opt_shape=[1, 3, 640, 640], | |
| max_shape=[1, 3, 1344, 1344]))) | |
| ]) | |
| use_efficientnms = False # whether to replace TRTBatchedNMS plugin with EfficientNMS plugin # noqa E501 | |