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

Celeb Dataset training #48

Closed
zhuzhen1996 opened this issue May 7, 2021 · 2 comments
Closed

Celeb Dataset training #48

zhuzhen1996 opened this issue May 7, 2021 · 2 comments
Assignees
Labels
enhancement New feature or request

Comments

@zhuzhen1996
Copy link

You have changed the two files train_binclass.py and split.py before. But notice that the two parameters related to celeb are not passed in the make_splits in split.py. I want to add training for the Celeb dataset in training.py. I would like to ask, do I need to modify other files if I add it directly to the parameter list? Can you edit and add it for me? There is another problem. It seems that you did not add the AUC accuracy parameter in train_triplet.py, and there is only the loss map in the obtained map.
image
image

@CrohnEngineer CrohnEngineer self-assigned this May 7, 2021
@CrohnEngineer CrohnEngineer added the enhancement New feature or request label May 7, 2021
@CrohnEngineer
Copy link
Collaborator

Hey @zhuzhen1996 ,

You have changed the two files train_binclass.py and split.py before. But notice that the two parameters related to celeb are not passed in the make_splits in split.py. I want to add training for the Celeb dataset in training.py

you're right.
We are still not 100% sure about including the complete training pipeline for Celeb-DF as we didn't use this dataset in the original paper, and this is a repository primarily meant to replicate the results of a scientific paper. I hope you can understand.
Anyway, I think we can discuss this internally and choose if it is the case to go full way with the support of Celeb-DF: in that case, I will warn you when the code is ready for pulling from the master.
If you can't wait, the only modifications you need to do are in the make_splits.py and load_df.py functions, adding the parameters for the Celeb-DF faces DataFrame and faces directories paths. It should be really easy to modify this functions (you just need to copy the code used for FF and DFDC).

It seems that you did not add the AUC accuracy parameter in train_triplet.py, and there is only the loss map in the obtained map.

That's right! We only map the loss as this is the primarily metric we use in all the paper.
Again, if you want to add this I think it would be trivial.
You are also welcome to open a PR for adding it as a functionality in the official repo :)
Bests,

Edoardo

@zhuzhen1996
Copy link
Author

zhuzhen1996 commented May 8, 2021 via email

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

No branches or pull requests

2 participants