2018-06-04 10:46:48 PSTTraceback (most recent call last):
2018-06-04 10:46:48 PSTFile "gan.py", line 296, in <module>
2018-06-04 10:46:48 PSTtrain_network()
2018-06-04 10:46:48 PSTFile "gan.py", line 267, in train_network
2018-06-04 10:46:48 PSTsave_images(imgtest, [8,8] ,new_image_path + '/epoch' + str(i) + '.jpg')
2018-06-04 10:46:48 PSTFile "/code/utils.py", line 34, in save_images
2018-06-04 10:46:48 PSTreturn imsave(inverse_transform(images), size, image_path)
2018-06-04 10:46:48 PSTFile "/code/utils.py", line 67, in imsave
2018-06-04 10:46:48 PSTimage = np.squeeze(merge(images, size))
2018-06-04 10:46:48 PSTFile "/code/utils.py", line 53, in merge
2018-06-04 10:46:48 PSTimg[j * h:j * h + h, i * w:i * w + w, :] = image
2018-06-04 10:46:48 PSTValueError: could not broadcast input array from shape (128,128,3) into shape (0,128,3)
As soon as finish the first epoch this occurs.
if i%50 == 0:
# save images
if not os.path.exists(new_image_path):
os.makedirs(new_image_path)
sample_noise = np.random.uniform(-1.0, 1.0, size=[batch_size, random_dim]).astype(np.float32)
imgtest = sess.run(fake_image, feed_dict={random_input: sample_noise, is_train: False})
# imgtest = imgtest * 255.0
# imgtest.astype(np.uint8)
save_images(imgtest, [8,8] ,new_image_path + '/epoch' + str(i) + '.png') # <<<< HERE