Skip to content

Instantly share code, notes, and snippets.

@AmrutaKoshe

AmrutaKoshe/predict.py

Last active July 19, 2021 07:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save AmrutaKoshe/8a3bca12bfb5320e8d9f1e72f0b6e344 to your computer and use it in GitHub Desktop.
Save AmrutaKoshe/8a3bca12bfb5320e8d9f1e72f0b6e344 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict.py
image=load_img("dog_photos/test/image4.jpg",target_size=(100,100))
image=img_to_array(image)
image=image/255.0
prediction_image=np.array(image)
prediction_image= np.expand_dims(image, axis=0)
prediction=model.predict(prediction_image)
value=np.argmax(prediction)
move_name=mapper(value)
print("Prediction is {}.".format(move_name))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment