-
Notifications
You must be signed in to change notification settings - Fork 1
/
example.py
44 lines (33 loc) · 931 Bytes
/
example.py
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
41
42
43
44
#!/usr/bin/env python
# Example usage of neuralart.py.
# Images can be converted to video with ffmpeg.
# > ffmpeg -pattern_type glob -i "*.png" -c:v huffyuv output.avi
from __future__ import print_function
import os
import sys
from PIL import Image
import neuralart
RENDER_SEED = 4
ITERATIONS = 200
RESOLUTION = 1024
Z_DIMS = 3
if len(sys.argv) != 2:
sys.stderr.write("Usage: {} DIRECTORY\n".format(sys.argv[0]))
sys.exit(1)
directory = sys.argv[1]
if not os.path.exists(directory):
os.makedirs(directory)
zfill = len(str(ITERATIONS - 1))
z = [-1.0] * Z_DIMS
step_size = 2.0 / ITERATIONS
for x in range(ITERATIONS):
result = neuralart.render(
xres=RESOLUTION,
seed=RENDER_SEED,
channels=3,
z=z
)
file = os.path.join(directory, str(x).zfill(zfill) + ".png")
im = Image.fromarray(result.squeeze())
im.save(file, 'png')
z = [_z + step_size for _z in z]