search

google-ai-edge / mediapipe

Cross-platform, customizable ML solutions for live and streaming media.
https://ai.google.dev/edge/mediapipe
Apache License 2.0
27.83k stars 5.18k forks source link

FaceMesh SavedModel Availability #2763

Closed nihalraman closed 2 years ago

nihalraman commented 3 years ago

Hi! Are the savedmodel/.pb files for Facemesh are publicly available? I'm looking to fine-tune the model for an individual person (similar to https://www.tensorflow.org/hub/tutorials/tf2_image_retraining), and I can only find .tflite and TensorFlow js format files online. Thanks for the help!

sgowroji commented 3 years ago

Hi @nihalraman, You can find the models and model cards from here https://google.github.io/mediapipe/solutions/models.html.

nihalraman commented 3 years ago

How do I convert either of these formats to the savedmodel format (or any other format where I can easily fine-tune the model with additional data)? Sorry if this is a basic question--I'm pretty new to this. Thanks for the help!

google-ml-butler[bot] commented 3 years ago

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] commented 3 years ago

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

google-ml-butler[bot] commented 3 years ago

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

sgowroji commented 2 years ago

Hello, unfortunately we do not plan to share the training code for FaceMesh models in the nearest future. If you would like to train your custom FaceMesh model, you might be interested in some details, on how we trained our existing models: https://arxiv.org/abs/1907.06724 and https://google.github.io/mediapipe/solutions/face_mesh.html

google-ml-butler[bot] commented 2 years ago

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