Skip to content

Instantly share code, notes, and snippets.

@Akash-Rawat

Akash-Rawat/Sample_Prediction

Last active July 2, 2021 10:42
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 Akash-Rawat/8c133d591ab621a935848bc5858d9741 to your computer and use it in GitHub Desktop.
Save Akash-Rawat/8c133d591ab621a935848bc5858d9741 to your computer and use it in GitHub Desktop.
Download ZIP
Sample_Prediction
Raw
Sample_Prediction
plt.imshow(images[4].to("cpu").permute(1, 2, 0))
plt.axis("off")
_ = plt.title(get_caption(model.generate_caption(images[4].unsqueeze(0))))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment