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

Hi, does this have a pre-training model weights? #5

Open
xiaomao19970819 opened this issue Mar 18, 2022 · 1 comment
Open

Hi, does this have a pre-training model weights? #5

xiaomao19970819 opened this issue Mar 18, 2022 · 1 comment

Comments

@xiaomao19970819
Copy link

Thank you for your work. Does this have a pre-training model weights? if i use my own dataset,is it accurate to use the Inceptionv3 model parameters pretrained on Imagenet to evaluate FID or IS ?
Should I change the "nclass_dict" and "classes_per_sheet_dict" values in the utils.py file to "3" when there are only three classes in my dataset.

@TDeVries
Copy link
Collaborator

We haven't released any pretrained model weights, but I think I may still have some lying around somewhere if you need them.

For GAN evaluation it is common practice to use an Inceptionv3 model pretrained on ImageNet to evaluate FID and IS. Although, it has been shown that this may bias the metric to favour the generation of samples that contain features indicative of ImageNet classes (https://arxiv.org/pdf/2203.06026.pdf). An alternative is to use a model trained in a self-supervised manner, such as https://github.com/stanis-morozov/self-supervised-gan-eval.

Yes, if you have your own dataset you'll likely need to update those values accordingly.

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