File size: 1,414 Bytes
47c8ba5 5360b85 47c8ba5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
# Projected GAN
- https://arxiv.org/abs/2111.01007
- https://github.com/autonomousvision/projected_gan
- weights
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/art_painting.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/church.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/bedroom.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/cityscapes.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/clevr.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/ffhq.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/flowers.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/landscape.pkl
- https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/pokemon.pkl
```python
import pathlib
import pickle
import sys
sys.path.insert(0, '~/codes/clones/projected_gan')
import dnnlib
import legacy
paths = sorted(pathlib.Path('orig'))
out_dir = pathlib.Path('models')
out_dir.mkdir()
for path in paths:
with dnnlib.util.open_url(path.as_posix()) as f:
model = legacy.load_network_pkl(f)['G_ema'].to('cpu')
ckpt = {'G_ema': model}
outpath = out_dir / path.name
with open(outpath, 'wb') as f:
pickle.dump(ckpt, f)
```
|