hrnet_w48_fp32_trt.py

Go to the documentation of this file.

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