mseitzer / pytorch-fid

Compute FID scores with PyTorch.
Apache License 2.0
3.22k stars 497 forks source link

Query: WGAN-GP FID SCORE (PyTorch) #89

Closed KomputerMaster64 closed 1 year ago

KomputerMaster64 commented 1 year ago

Thank you for sharing the implementations of the GAN based models on popular datasets like CelebA. I have implemented the WGAN-GP model (in PyTorch), the samples are looking closer the reported work (please refer to the attached image). When I try to evaluate the Fréchet inception distance (FID score), I am not able to comprehend the high values of 100+ (best value 113.4). Others have reported lower FID Scores. Authors in the Quality Aware Generative Adversarial Networks compared the vairous FID scores for various GAN models and there Ishaan Gulrajani's official implementation of WGAN-GP got FID score of 12.89.

I request you to guide me. Regards Prabhav

I have used the following repositories for reference while implementing the WGAN-GP model and evaluating the FID scores: LynnHo/DCGAN-LSGAN-WGAN-GP-DRAGAN-Pytorch CharlesNord/WGAN-GP-DRAGAN-Celeba-Pytorch eriklindernoren/PyTorch-GAN hukkelas/pytorch-frechet-inception-distance mseitzer/pytorch-fid

mseitzer commented 1 year ago

This may have various causes unrelated to this library, and I can not answer general support requests.

As general advise, FID is sensitive to details related to how images are created and stored. Thus in order to compare with FID scores reported in other works, you need to make sure to reproduce the way they computed FID.

Some ideas for you to try for the case of CelebA: