BicycleGAN icon indicating copy to clipboard operation
BicycleGAN copied to clipboard

Can this model train with gray scale code?

Open Github-Vicente opened this issue 6 years ago • 1 comments

The code is concise and worked great. But when I change the batch size to 30 in main.py. It will show File "/home/BicycleGAN/model.py", line 261, in train _, summary_str_d, D_loss_curr = self.sess.run([self.D_solver, self.d_loss_sum, self.loss_D], feed_dict={self.image_A: batch_imagesA, self.image_B: batch_imagesB, self.z: batch_z}) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 887, in run run_metadata_ptr) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1086, in _run str(subfeed_t.get_shape()))) ValueError: Cannot feed value of shape (1, 256, 256, 3) for Tensor u'input_images:0', which has shape '(30, 256, 256, 3)' Is there anything else that I need to revise?

Thank you very much.

Github-Vicente avatar Dec 01 '18 10:12 Github-Vicente

@GamuzaLu ...You can't train with batch_size = 30 using existing code. The code uses batch_size = 1 and you are passing 30 images in a batch. See function load_batch_image in load_data.py file which returns a single pair of an image.

prakashpandey9 avatar Dec 04 '18 19:12 prakashpandey9