mediapipe icon indicating copy to clipboard operation
mediapipe copied to clipboard

MediaPipe Pose - add training data to the model?

Open tomerlevi444 opened this issue 2 years ago • 7 comments

Is it possible to add training data to MediaPipe Pose, so it better recognizes irregular poses?

tomerlevi444 avatar Mar 27 '22 22:03 tomerlevi444

Hi @tomerlevi444 , Did you get a chance to check this closed issue https://github.com/google/mediapipe/issues/1824

sureshdagooglecom avatar Mar 28 '22 06:03 sureshdagooglecom

What you suggest is not quite what I look for.

I have tens of thousands of annotated images of a person in different body poses, and I would like to train the tflite model of mediapipe to learn (not predict) from these images.

tomerlevi444 avatar Mar 28 '22 08:03 tomerlevi444

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] avatar Apr 04 '22 08:04 google-ml-butler[bot]

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

google-ml-butler[bot] avatar Apr 11 '22 09:04 google-ml-butler[bot]

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

google-ml-butler[bot] avatar May 25 '22 06:05 google-ml-butler[bot]

Hey, in case you're still interested, check out #3410 , #2761 and #507. I have been looking for an answer on this as well and found these threads quite helpful.

Cleoxx avatar Aug 06 '22 22:08 Cleoxx

Hi @tomerlevi444 , For python you can create your own graph using the calculators and execute as mentioned here https://google.github.io/mediapipe/getting_started/python_framework.html#graph. Have you looked at these closed issues https://github.com/google/mediapipe/issues/2761 and https://github.com/google/mediapipe/issues/507. Thank you!

sureshdagooglecom avatar Sep 02 '22 06:09 sureshdagooglecom

Hi @tomerlevi444, I assume you intended make the existing Mediapipe model learn from your data (transfer learning) than training a model from scratch. Hope you found a solution to the problem you mentioned.

Were you able to use the Mediapipe model and make it learn from your own dataset? I am having a similar requirement and unable to find a solution.

csn1800 avatar Nov 01 '22 13:11 csn1800

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] avatar Nov 22 '22 07:11 google-ml-butler[bot]

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

google-ml-butler[bot] avatar Nov 29 '22 07:11 google-ml-butler[bot]

Hi @tomerlevi444 and @csn1800 did either of you have any success with transfer learning with Mediapipe? I'm trying to find if this is possible myself.

il-nietos avatar Mar 15 '23 08:03 il-nietos

@il-nietos I have not given much time for transfer learning later on, however, it seems that PINTO mode zoo maybe of help to you.

For eg, you may use https://github.com/PINTO0309/PINTO_model_zoo/tree/main/053_BlazePose, download the resources and try using the saved model to enhance and/or modify the model further. I have not tried it on my own, so not sure how efficient the models can be on resource constrained platforms.

csn1800 avatar Mar 15 '23 13:03 csn1800

Objective: To Fine Tune Pose Estimation Solution with Animal Dataset.

I want to use Mediapipe Solution (POSE) for animal pose estimation. I have the raw data of cow pictures, how to make that data compatible to be trained with Mediapipe Solution?

zaidworks515 avatar Nov 20 '23 12:11 zaidworks515