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
Add methods for model training #7
Comments
How to train on my own custom dataset |
@shaikhaman786vector thank you for the prompt, I've been dragging my feet on getting this in the repo. I've just updated the repo and python package to include some utilities for finetuning the models. If you want to train on your own dataset, first update the package to the latest version on PyPI (0.2.2): pip install --upgrade facenet-pytorch After running the above command, follow this notebook to finetune a model on your own dataset. Depending on your dataset you may need to alter training parameters (e.g., learning rate, optimizer, weight decay) or data augmentation (e.g., random resizing, random horizontal flip, etc.). If you get a chance, please let me know if you get the model training well. The finetuning notebook works well for me, but hasn't been tested much. |
it will be good if you upload for triplet loss |
Also if there will be code for recognizing new image, this will be complete flow for face recognition |
@shaikhaman786vector I'm not sure what you mean by "recognizing new image". Do you mean an image for a person that was in the training set (in which case you can just pass the image tensor to the model), or an image of a new person? There is an example showing how to infer using a trained model in the repo (examples/infer.ipynb). |
The repo currently only includes models (MTCNN and InceptionResnetV1). It would be good to add code for updating pretrained models given new data or different training hyperparameters.
The text was updated successfully, but these errors were encountered: