Replace pose_detection.tflite with my own model.

[mch0dmin]

If I want to replace pose_detection.tflite with my own pose detection model(eg. HRNet-Human-Pose-Estimation), what should I do to make it work?

Can you give me some advice？

Thanks very much.

[sureshdagooglecom]

Hi @mch0dmin , could you elaborate this issue to understand better.

[mch0dmin]

hi @sureshdagooglecom ,

Glad to receive your reply！

I want to do this :

(1) First, I have a model of human pose estimation, eg. https://github.com/openvinotoolkit/open_model_zoo/tree/master/models/public/human-pose-estimation-3d-0001 or https://github.com/HRNet/HRNet-Human-Pose-Estimation.

(2) And then I convert these pose estimation models to .tflite format, eg. hrnet.tflite. ( Because the example in mediapipe uses the .tflite model.)

(3) Now i want to run mediapipe/examples/desktop/pose_tracking:pose_tracking_cpu, but i don't use default model pose_detection.tflite, which is provided by mediapipe. Instead, use the model I converted to replace it.

I have made some attempts, but I have encountered mistakes. I don't know whether this is feasible, or what model needs to be used to replace the model provided by mediapipe.

Thank you again for your reply.

[mch0dmin]

hi @sureshdagooglecom, can you give me some advice? Thanks!

[guker]

hi @sureshdagooglecom, can you give me some advice? Thanks!

pose_detection.tflite predict a box and four keypoint, do your model can predict a box and four keypoint? if not, your model cant be used.

[sureshdagooglecom]

Hi @mch0dmin , Models detects box and four keypoint.

[mch0dmin]

hi @guker and @sureshdagooglecom ， Thank you very much for your answers. Is there any way that I can get a similar model?

[sureshdagooglecom]

Hi @mch0dmin , We don't provide similar models .

[google-ml-butler[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you.

[google-ml-butler[bot]]

Closing as stale. Please reopen if you'd like to work on this further.

[google-ml-butler[bot]]

Are you satisfied with the resolution of your issue? Yes No