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

Fine-tuning? #2

Open
jrosebr1 opened this issue May 27, 2019 · 2 comments
Open

Fine-tuning? #2

jrosebr1 opened this issue May 27, 2019 · 2 comments

Comments

@jrosebr1
Copy link

Really nice job, @ksachdeva! Congratulations on this implementation -- it's really nice!

I was curious about fine-tuning with Keras. For example, let's say we wanted to:

  1. Take the dlib model definition and trained weights
  2. Convert them to Keras
  3. And then use Keras to fine-tune a model on data the original model wasn't originally trained on

Have you experimented at all with that use case?

I'd really love to help with such a project so please do get back to me 😄

@ksachdeva
Copy link
Owner

ksachdeva commented May 27, 2019 via email

@jrosebr1
Copy link
Author

For face recognition, yes, you could take the 128-d embeddings from the network and then train a Logistic Regression or SVM on top of those representations. That can and will work for a small number of new face identities to recognize.

To make the method more robust; however, one could fine-tune the model on a new dataset of example images. This new dataset would be smaller both in terms of (1) total images and (2) total number of unique individuals. It may also be impossible to train such a network from scratch using the dataset.

There is an "in-between" situation where the SVM/LR approach could be too noisy/too many incorrect labelings while training from scratch would be impossible. In those situations fine-tuning might be worth exploring (at least in my opinion).

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