Add files
Browse files- README.md +22 -0
- models/bicycles_256_pytorch.pkl +3 -0
- models/dogs_1024_pytorch.pkl +3 -0
- models/elephants_512_pytorch.pkl +3 -0
- models/giraffes_512_pytorch.pkl +3 -0
- models/horses_256_pytorch.pkl +3 -0
- models/lions_512_pytorch.pkl +3 -0
- models/parrots_512_pytorch.pkl +3 -0
README.md
CHANGED
@@ -9,3 +9,25 @@
|
|
9 |
- https://storage.googleapis.com/self-distilled-stylegan/giraffes_512_pytorch.pkl
|
10 |
- https://storage.googleapis.com/self-distilled-stylegan/parrots_512_pytorch.pkl
|
11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
- https://storage.googleapis.com/self-distilled-stylegan/giraffes_512_pytorch.pkl
|
10 |
- https://storage.googleapis.com/self-distilled-stylegan/parrots_512_pytorch.pkl
|
11 |
|
12 |
+
```python
|
13 |
+
import pathlib
|
14 |
+
import pickle
|
15 |
+
|
16 |
+
sys.path.insert(0, '~/codes/clones/stylegan3')
|
17 |
+
|
18 |
+
paths = sorted(pathlib.Path('orig/').glob('*'))
|
19 |
+
|
20 |
+
out_dir = pathlib.Path('models')
|
21 |
+
out_dir.mkdir()
|
22 |
+
|
23 |
+
for path in paths:
|
24 |
+
with open(path, 'rb') as f:
|
25 |
+
ckpt = pickle.load(f)
|
26 |
+
for key in list(ckpt.keys()):
|
27 |
+
if key != 'G_ema':
|
28 |
+
del ckpt[key]
|
29 |
+
out_path = out_dir / path.name
|
30 |
+
with open(out_path, 'wb') as f:
|
31 |
+
pickle.dump(ckpt, f)
|
32 |
+
```
|
33 |
+
|
models/bicycles_256_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7e05d9a4862f364cb8ac175256e63395f48d7a67abaadd21e68f5a1a459b0094
|
3 |
+
size 120932695
|
models/dogs_1024_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a882a83ad273db2b3401dd151dbd9108650108f9d1e61876c93e7b2431fd33e3
|
3 |
+
size 132775709
|
models/elephants_512_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ffee41bcdf4d5c34002c8ff44dea8fd829e53c371d5783754d4e410144a7a042
|
3 |
+
size 124006016
|
models/giraffes_512_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:25d67d97d4da0abfae82932ef42717ed3ba8a0e7baac444d88674e82e74d0393
|
3 |
+
size 124006016
|
models/horses_256_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9faf0dafc0fde9b89b7767d803952b202a3cd4f931d00ce0ed6d3984f40a0e8c
|
3 |
+
size 120932695
|
models/lions_512_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:14f1d291b916e8ef95ebde5c7d2c3e7c8e6f66e7690a3f35b245a86037046c34
|
3 |
+
size 124006016
|
models/parrots_512_pytorch.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e686940d4c006dd88b93963a494a97d50aaab60bbe2e152b9417b1b70ee9db87
|
3 |
+
size 124006016
|