ValueError: could not broadcast input array from shape (96,96,4) into shape (96,96)

I’m having the similar trouble as in #26. Not sure if I’m doing something wrong or if there’s a fix. I’m using my own image dataset all scaled to 96x96. Stack trace below.

Image: PNG 96x96 96x96+0+0 8-bit sRGB 7.2KB 0.000u 0:00.000

Traceback (most recent call last):
  File "main.py", line 99, in <module>
    tf.app.run()
  File "/home/ubuntu/.local/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 44, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "main.py", line 82, in main
    dcgan.train(FLAGS)
  File "/home/ubuntu/GANs/model.py", line 186, in train
    sample_inputs = np.array(sample).astype(np.float32)

[Zhangtd]

I had the same problem before. It is because there are grey scale images in my data. I solved this by:

from PIL import Image
img = Image.open(img_name).convert('RGB')
# your own image operations 

Simply converting grey scale image to RGB image is OK.

[empireshades]

I was able to fix this issue by:

1. Ensuring that all images in my dataset were the exact same size.

cd data/mypics
# Create dir to store images that aren't the same size`
mkdir misshaped
# Identify misshaped images using Imagemagick's 'identify' tool 
and move to above dir. (Replace with your desired resolution)
identify * | grep -v "600x450" | awk '{ print $1 }' | xargs -I {} bash -c "mv {} misshaped"

1. Ensuring that all images had the same number of colors. In my case I was using color JPGs and I found a handful of Grayscale images.

identify -format "%i %[colorspace]\n" IMG_*.jpg | grep -v sRGB
IMG_4959.jpg Gray
IMG_4960.jpg Gray
IMG_4961.jpg Gray
IMG_4962.jpg Gray
IMG_7356.jpg Gray
IMG_7630.jpg Gray

After I took care of these, it worked.