Skip to content

Instantly share code, notes, and snippets.

@VedPDubey

VedPDubey/custom_predictions4.py

Created July 13, 2021 17:56
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 VedPDubey/9cbfa49c7092ef5f15e2049324eb0f6b to your computer and use it in GitHub Desktop.
Save VedPDubey/9cbfa49c7092ef5f15e2049324eb0f6b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
custom_predictions4.py
def custom_predictions(path):
img = ig.load_img(path, target_size=(64, 64))
plt.imshow(img)
img = np.expand_dims(img, axis=0)
img.reshape(1,64,64,3)
prediction = np.argmax(model.predict(img))
plt.title(labels[prediction])
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment