Commit 857e3589 authored by Amelie Royer's avatar Amelie Royer

Adding verbose output for demo mode

parent 9dfd3ba1
......@@ -520,8 +520,9 @@ with tf.Session() as sess:
x_gen = np.zeros((1, WIDTH // args.downsample, HEIGHT // args.downsample, 3), dtype=float)
feed = ({x_gray_gen: image, scale_var:float(not args.sample_mode)})
feed.update({embedding_cache[i]: sess.run(embedding_cache[i], {x_gray_gen: image}) for i in range(args.nr_gpus)})
for yi in range(0, WIDTH // args.downsample):
print('\rProcessing pixel row %d/%d' % (yi + 1, WIDTH // args.downsample), end='')
for xi in range(0, HEIGHT // args.downsample):
feed.update({x_canvas_gen: x_gen})
new_x_gen_np = np.concatenate(sess.run(samplers_from_pic, feed))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment