microsoft / computervision-recipes

Best Practices, code samples, and documentation for Computer Vision.
MIT License
9.34k stars 1.16k forks source link

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

Open ShreyasJoshi7 opened 5 months ago

ShreyasJoshi7 commented 5 months ago

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:

review-notebook-app[bot] commented 5 months ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

ShreyasJoshi7 commented 5 months ago

@microsoft-github-policy-service agree

smith-kyle commented 5 months ago

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 commented 4 months ago

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 commented 4 months ago

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

ShreyasJoshi7 commented 3 months ago

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.