File size: 575 Bytes
c508d7f |
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 |
#!/usr/bin/env python
import os
import numpy as np
from PIL import Image
import chainer
import chainer.cuda
from chainer import Variable
def out_generated_image(gen, dis, rows, cols, seed, dst, writer):
@chainer.training.make_extension()
def make_image(trainer):
np.random.seed(seed)
n_images = rows * cols
xp = gen.xp
z = Variable(xp.asarray(gen.make_hidden(n_images)))
with chainer.using_config('train', False):
x = gen(z)
writer.add_image('img', x, trainer.updater.iteration)
return make_image
|