rtmdet_trt.py

Go to the documentation of this file.

model = dict(
    type='MMDetTRTRunner',
    deploy_cfg = '/workspace/mmdeploy/configs/mmdet/instance-seg/instance-seg_rtmdet-ins_tensorrt_fp16_static-640x640.py',
    model_cfg = '/workspace/mmdetection/configs/rtmdet/rtmdet-ins_x_8xb16-300e_coco.py',
    device = 'cuda',
    backend_model = ['/workspace/mmdeploy_model/rtmdet/end2end.engine'],
    )