Skip to content

Instantly share code, notes, and snippets.

@pranjalAI

pranjalAI/inference.py

Last active September 8, 2020 14:40
Show Gist options
  • Save pranjalAI/56631da1cb58e77c931bb7a1b716be97 to your computer and use it in GitHub Desktop.
Save pranjalAI/56631da1cb58e77c931bb7a1b716be97 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
inference.py
test_image=cv2.imread("IMAGE_PATH")
test_image = np.expand_dims(test_image, axis=0)
out=np.argmax(model_loaded.predict(test_image))
Ans=np.where(out==0, "Middle",
(np.where(out==1,"Old",
"Young"))).item()
print(Ans+" age person.")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment