TensorFlow Image Classifier Returns The Same Label For Each Image

I have made an image classifier to classify images of airplanes and rockets based on the tutorial at https://blog.francium.tech/build-your-own-image-classifier-with-tensorflow-and-

Solution 1:

The bug is caused by this line, if np.argmax(model_out == 1):. Please change it to if np.argmax(model_out) == 1:. In the blog you post, they are the second one.