How should I get a heatmap output from a converted mmpose model?

[starashima]

Hi. I could successfully get a keypoint output from a converted mmpose model (i.e., end2end.engine generated from the default HRNet-W48 config ), but I also want to get a corresponding heatmap output as I can get using mmpose api.
How should I do this?

Thank you very much for your help.

[lvhan028]

@RunningLeon

[RunningLeon]

@starashima Hi, actually the trt engine only outputs heatmap. the decoding part is added in the End2EndModel. You can directly get the heatmap in here.

mmdeploy/mmdeploy/codebase/mmpose/deploy/pose_detection_model.py

Line 95 in 4d4c10a

heatmaps = outputs[0]