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

added image classification using transformer feature notebook #689

Closed
wants to merge 1 commit into from

Conversation

ShreyasJoshi7
Copy link

Description

I have added a pynb which contains inference code for image classification using transformer. The pipeline provides a checkpoint which can be downloaded and used for adding new classes and fine tuning, a test image and the a json file of ImageNet classes. The user need not worry about the required files. The code downloads all the required file from my S3 bucket. This is a pulg-and-play notebook.

This change is required as an issue is raised to include an image classification model which utilises transformer.
This is the issue #raied:
**[FEATURE_REQUEST] Add vision transformers model to image classification #678 **

Related Issues

[FEATURE_REQUEST] Add vision transformers model to image classification #678
https://github.com/microsoft/computervision-recipes/issues/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

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

1 participant