8000 GitHub - sangkv/ViT: ViT Transfer Learning for Image Classification
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

sangkv/ViT

Repository files navigation

ViT Transfer Learning for Image Classification

This project demonstrates how to perform image classification using transfer learning with the ViT architecture.

alt text

Installation

Install all dependencies:

pip install -r requirements.txt

Transfer Learning

python transfer-learning.py

Export the Model to TorchScript

python compile.py

Demo

To run a demo and see the model in action:

python demo.py

Contact

For questions or collaboration, feel free to contact:

sangkv.work@gmail.com

About

ViT Transfer Learning for Image Classification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

0