I've images array which appends processed images one by one
tmp = image[:,:,1].reshape(160, 320, 1)
images.append(tmp)
After that I'm converting images to numpy array
X_train = np.array(images)
But, I'm getting
ValueError: could not broadcast input array from shape (160,320,1) into shape (160,320)