microsoft / computervision-recipes

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

added image classification using transformer feature notebook #689

Closed ShreyasJoshi7 closed 10 months ago

ShreyasJoshi7 commented 10 months ago

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:

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

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB