hrnet_trt.py

Go to the documentation of this file.

model = dict(
    type='MMPoseTRTRunner',
    deploy_cfg = '/workspace/mmdeploy/configs/mmpose/pose-detection_tensorrt-fp16_static-256x192.py',
    model_cfg = '/workspace/mmdeploy/td-hm_hrnet-w32_8xb64-210e_coco-256x192.py',
    device = 'cuda',
    backend_model = ['/workspace/mmdeploy_model/hrnet_pose/end2end.engine'],
    batch_size=1,
    )