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)
```