def sample(self, path, save_samples):
gan = self.gan
generator = gan.uniform_sample
z_t = gan.uniform_encoder.sample
x_t = gan.inputs.x
sess = gan.session
config = gan.config
global x_v
global z_v
x_v = sess.run(x_t)
x_v = np.tile(x_v[0], [gan.batch_size(),1,1,1])
sample = sess.run(generator, {x_t: x_v})
stacks = []
bs = gan.batch_size()
width = 5
print(np.shape(x_v), np.shape(sample))
stacks.append([x_v[1], sample[1], sample[2], sample[3], sample[4]])
for i in range(bs//width-1):
stacks.append([sample[i*width+width+j] for j in range(width)])
images = np.vstack([np.hstack(s) for s in stacks])
self.plot(images, path, save_samples)
return [{'images': images, 'label': 'tiled x sample'}]
评论列表
文章目录