zideliu/StyleDrop-PyTorch

python extract_empty_feature.py

wpl427 opened this issue · 1 comments

wpl427 commented

.
├── assets
│ ├── ckpts
│ │ ├── cc3m-285000.ckpt
│ │ │ ├── lr_scheduler.pth
│ │ │ ├── nnet_ema.pth
│ │ │ ├── nnet.pth
│ │ │ ├── optimizer.pth
│ │ │ └── step.pth
│ │ └── imagenet256-450000.ckpt
│ │ ├── lr_scheduler.pth
│ │ ├── nnet_ema.pth
│ │ ├── nnet.pth
│ │ ├── optimizer.pth
│ │ └── step.pth
│ ├── fid_stats
│ │ ├── fid_stats_cc3m_val.npz
│ │ └── fid_stats_imagenet256_guided_diffusion.npz
│ ├── pipeline.png
| ├── contexts
│ │ └── empty_context.npy
└── └── vqgan_jax_strongaug.ckpt
python extract_empty_feature.py
The fid stats folder and the following two files are missing

.
├── assets
│ ├── ckpts
│ │ ├── cc3m-285000.ckpt
│ │ │ ├── lr_scheduler.pth
│ │ │ ├── nnet_ema.pth
│ │ │ ├── nnet.pth
│ │ │ ├── optimizer.pth
│ │ │ └── step.pth
│ │ └── imagenet256-450000.ckpt
│ │ ├── lr_scheduler.pth
│ │ ├── nnet_ema.pth
│ │ ├── nnet.pth
│ │ ├── optimizer.pth
│ │ └── step.pth
│ ├── pipeline.png
| ├── contexts
│ │ └── empty_context.npy
└── └── vqgan_jax_strongaug.ckpt

Download the reference statistics for FID from this link. Place the downloaded .npz file in assets/fid_stats.

But actually, they are not used