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

train new feature classifier on face dataset problem #15

Closed
shartoo opened this issue Aug 6, 2019 · 2 comments
Closed

train new feature classifier on face dataset problem #15

shartoo opened this issue Aug 6, 2019 · 2 comments

Comments

@shartoo
Copy link

shartoo commented Aug 6, 2019

I'm trying to train a face feature classifier and then get a feature axis like TL-GAN ,but it seems that the dlib face detector did not work very well,it may fail on some facial image. I checked some face detection model and thought MCTNN may be a better replacement for dlib in perceptual_model.py ,as MTCNN is of higher accuracy and faster.

@shartoo
Copy link
Author

shartoo commented Aug 6, 2019

And for fined face mask,there is a kind of algorithm names image matting, repo like global matting may helps but it need tri-map mask(a rough mask which can be generated by your morphology cv algorithm). Thank you for your sharing!

@pbaylies
Copy link
Owner

pbaylies commented Aug 7, 2019

@shartoo the main reason to use dlib here is just because NVIDIA used it in training their pre-trained models in the first place, and provided their implementation of aligning images, so we can align them in a consistent fashion. But really any detector that provides the same basic landmarks should be fine, and would allow you to more or less replicate their results.

Also, it doesn't matter so much if you're training your own models, in that case you can process your data in a consistent fashion so that the encoded images are treated in the same way. I'm glad you found the masking helpful, grabcut was designed for it, really. If you'd like to provide another compatible method for aligning images and/or generating masks, well, patches are welcome!

@pbaylies pbaylies closed this as completed Aug 7, 2019
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

2 participants