Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

hope support mobilenetv3 small #28

Closed
jiangjianping opened this issue Dec 30, 2020 · 16 comments
Closed

hope support mobilenetv3 small #28

jiangjianping opened this issue Dec 30, 2020 · 16 comments

Comments

@jiangjianping
Copy link

hello Floe,

Could you tell me how to add the support of mobilenetv3 small?

@floe
Copy link
Owner

floe commented Dec 31, 2020

Not quite sure what you mean, deeplab v3 is already based on mobilenet v3? Or do you have a link for the specific network you want to support?

@jiangjianping
Copy link
Author

@floe
Copy link
Owner

floe commented Jan 2, 2021

Ah, interesting. First thing you would need is the actual model file in .tflite format, apparently it's not available in https://github.com/google/mediapipe/tree/master/mediapipe/models ?

@jiangjianping
Copy link
Author

@floe
Copy link
Owner

floe commented Jan 3, 2021

Deeplab v3 is exactly what my project here already uses: https://github.com/floe/deepbacksub/blob/master/deeplabv3_257_mv_gpu.tflite

The only difference I can see is that the one you just posted is trained on Cityscapes, which won't help a lot with person segmentation.

The one from Google Meet would be much more interesting, because it is specifically optimized for person segmentation, but you would need to find the specific .tflite file for that Google Meet network.

@jiangjianping
Copy link
Author

Floe,
Thanks, Hope lucky to get that tfilte. Do you have idea about joint bilateral filter?

@floe
Copy link
Owner

floe commented Jan 3, 2021

Right now, it seems you can download the models from https://meet.google.com/_/rtcvidproc/release_1wttl/345264209/segm_full_v679.tflite and https://meet.google.com/_/rtcvidproc/release_1wttl/345264209/segm_lite_v681.tflite - and according to the model card they should be licensed with Apache 2.0, so you should in theory be allowed to re-use them. I'll give it a try and see if these can be integrated.

@jiangjianping
Copy link
Author

Floe,
Great! Thanks a lot.

@jiangjianping
Copy link
Author

Hello Floe,
I tried the new model, but the AllocateTensors() failed. I guess that the xnnpack need to be included.

@jiangjianping
Copy link
Author

Got something may be useful

tensorflow/tfjs#4177

@floe
Copy link
Owner

floe commented Jan 4, 2021

I've just copied the relevant custom op from MediaPipe for now. See latest commit (306da1b) for a implementation that's somewhat working (still some sort of synchronization error between capture thread and network evaluation happening here, though).

@jiangjianping
Copy link
Author

Great! I will try.

@floe
Copy link
Owner

floe commented Jan 4, 2021

I'm closing this for now, thanks for pointing me to the Google Meet segmentation model. Please open a new issue if something is still wonky.

@fransiskusyoga
Copy link

Hello, I am trying to run google meet model on windows. But I still don't have any idea how to integrate the custom op from mediapipe. I cannot load the model using normal runtime because it doesn't have the custom operation.

@floe
Copy link
Owner

floe commented Feb 2, 2021

@fangfufu
Copy link

fangfufu commented Jun 8, 2021

Mediapipe officially released selfie segmentation support. You might want to update your app.
https://github.com/google/mediapipe/releases/tag/v0.8.5

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants