(400, 400)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-3-8359524d557b> in <module>()
     19     print(image.shape)
     20     #training_data[i] = image.reshape(-1)
---> 21     training_data[i] = image
     22
     23     i+=1

ValueError: could not broadcast input array from shape (400,400) into shape (400,400,3)