i72sijia commited on
Commit
b76dc12
1 Parent(s): 8131ffc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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)