Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,12 +25,12 @@ def seed2vec(Gs, seed):
|
|
25 |
def init_random_state(Gs, seed):
|
26 |
rnd = np.random.RandomState(seed)
|
27 |
noise_vars = [var for name, var in Gs.components.synthesis.vars.items() if name.startswith('noise')]
|
28 |
-
tflib.set_vars({var: rnd.randn(*var.shape.as_list()) for var in noise_vars}) # [height, width]
|
29 |
|
30 |
def generate_image(Gs, z, truncation_psi, prefix="image", save=False, show=False):
|
31 |
# Render images for dlatents initialized from random seeds.
|
32 |
Gs_kwargs = {
|
33 |
-
'output_transform': dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True),
|
34 |
'randomize_noise': False
|
35 |
}
|
36 |
if truncation_psi is not None:
|
@@ -131,7 +131,7 @@ def concatenate(img_array):
|
|
131 |
def initiate(seed, n_imgs, text):
|
132 |
|
133 |
pkl_file = "networks/experimento_2.pkl"
|
134 |
-
tflib.init_tf()
|
135 |
|
136 |
with open(pkl_file, 'rb') as pickle_file:
|
137 |
_G, _D, Gs = pickle.load(pickle_file)
|
|
|
25 |
def init_random_state(Gs, seed):
|
26 |
rnd = np.random.RandomState(seed)
|
27 |
noise_vars = [var for name, var in Gs.components.synthesis.vars.items() if name.startswith('noise')]
|
28 |
+
dnnlib.tflib.set_vars({var: rnd.randn(*var.shape.as_list()) for var in noise_vars}) # [height, width]
|
29 |
|
30 |
def generate_image(Gs, z, truncation_psi, prefix="image", save=False, show=False):
|
31 |
# Render images for dlatents initialized from random seeds.
|
32 |
Gs_kwargs = {
|
33 |
+
'output_transform': dict(func=dnnlib.tflib.convert_images_to_uint8, nchw_to_nhwc=True),
|
34 |
'randomize_noise': False
|
35 |
}
|
36 |
if truncation_psi is not None:
|
|
|
131 |
def initiate(seed, n_imgs, text):
|
132 |
|
133 |
pkl_file = "networks/experimento_2.pkl"
|
134 |
+
dnnlib.tflib.init_tf()
|
135 |
|
136 |
with open(pkl_file, 'rb') as pickle_file:
|
137 |
_G, _D, Gs = pickle.load(pickle_file)
|