Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fixing issue #678 -> pynb for image classification using vision transformer pretrained on ImageNet. #690

Open
wants to merge 9 commits into
base: staging
Choose a base branch
from

Conversation

ShreyasJoshi7
Copy link

Description

I have added a notebook which contains inference code for image classification using vision transformer pretrained on IamgeNet. The code performs the following tasks:

  1. downloads the checkpoint (which can be downloaded and used to fintune the model on aditional classes).
  2. downloads a test image.
  3. downloads the json file of ImageNet classes.

This notebook solves the issue #678 which requires an image classifier which uses transformer.

Related Issues

[FEATURE_REQUEST] Add vision transformers model to image classification #678

Checklist:

  • [ x] I have followed the contribution guidelines and code style for this project.
  • [ x] This branch is created from staging and not master.
  • [ x] This PR is being made to staging and not master.
  • [ x] I will squash merge this PR into staging.
  • [ x] I have added tests covering my contributions.
  • [ x] I have updated the documentation accordingly.

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@ShreyasJoshi7
Copy link
Author

@microsoft-github-policy-service agree

@smith-kyle
Copy link

Just a heads up, you can review these notebook changes with GitNotebooks too: https://gitnotebooks.com/microsoft/computervision-recipes/pull/690

I think it offers some advantages over what you're currently using.

@ShreyasJoshi7
Copy link
Author

Just a heads up, you can review these notebook changes with GitNotebooks too: https://gitnotebooks.com/microsoft/computervision-recipes/pull/690

I think it offers some advantages over what you're currently using.

@smith-kyle thank you for the suggestion.

@ShreyasJoshi7
Copy link
Author

My Amazon account expired and I had to update the S3 link.

@ShreyasJoshi7
Copy link
Author

I am note able to find the information about the author, but can the author please merge this PR? It has passed all the checks.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants