How to speed up the fid?

GaParmar / clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]

https://www.cs.cmu.edu/~clean-fid/

MIT License

894 stars 68 forks source link

How to speed up the fid? #31

Open wingvortex opened 2 years ago

wingvortex commented 2 years ago

Hi, thanks for sharing your work. In my case, I have a reference folder (fdir1), and about 50 target folders (fdir2), each containing hundreds of images. It takes a long time to calculate the score by default fid.compute_fid(fdir1, fdir2). It seems it is chocked by the CPU. Is there any way to speed up it?

GaParmar commented 2 years ago

Hi, yes this should not take too long. Can you specify what are the flags that you are using when computing the FID?

-Gaurav

yuxu915 commented 1 year ago

Hi, yes this should not take too long. Can you specify what are the flags that you are using when computing the FID?

-Gaurav

hi, I have the same problem. My command is as : "fid_score = fid.compute_fid('fold1','fold2',mode='clean')"