In this video, we will train a Vision Transformer (ViT) model for classifying images of vehicles into multiple categories using PyTorch! 🔥
We will utilizes the 'google/vit-base-patch16-224-in21k' model from Hugging Face, fine-tuning it on a custom dataset.
You will learn:
✅ Loading and transforming image datasets
✅ Fine-tune a VIT model (vision transformer)
✅ Training and optimizing the ViT model (vision transformer)
✅ Test our model with a test dataset , and predict the result
Code for the tutorial : https://ko-fi.com/s/e37ef6db80
You can find more computer vision tutorials in my blog page : https://eranfeit.net/blog/
You can find more Visual Language models tutorials tutorials in this playlist : https://www.youtube.com/playlist?list=PLdkryDe59y4a2PRJda-Z7M7Sod7uQKT2d
You can find more image classification tutorials in this playlist : https://www.youtube.com/watch?v=8k6oNjl2EgE&list=PLdkryDe59y4aytIPjci6_fn3B1-QuM-Oh
~~~~~~~~~~~~~~~ recommended courses and books ~~~~~~~~~~~~~~~
A perfect course for learning modern Computer Vision with deep dive in TensorFlow , Keras and Pytorch . You can find it here : http://bit.ly/3HeDy1V
I also recommend this book, https://amzn.to/44GnlLW : "Make Your Own Neural Network - An In-depth Visual Introduction For Beginners ".
~~~~~~~~~~~~~~~ CONNECT ~~~~~~~~~~~~~~~
☕ Buy me a coffee - https://ko-fi.com/eranfeit
🖥️ Email : feitgemel@gmail.com
🌐 https://eranfeit.net
🤝 Fiverr : https://www.fiverr.com/s/mB3Pbb
🐦 Twitter - https://twitter.com/eran_feit
📸 Instagram - https://www.instagram.com/eran_feit/
▶️ Subscribe - youtube.com/@eranfeit?sub_confirmation=1
🐙 Facebook - https://www.facebook.com/groups/3080601358933585
📝 Medium - https://medium.com/@feitgemel
~~~~~~~~~~~~~~ SUPPORT ME 🙏~~~~~~~~~~~~~~
🅿 Patreon - https://www.patreon.com/EranFeit
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
00:00 Introduction
03:31 Installation
06:51 Discover the dataset
08:27 Fine-tune the VIT model
25:33 Test the model (Prediction)
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#EranFeit #imageclassification #visiontrasformers
~~~~~~~~~~~~~~ Credits ~~~~~~~~~~~~~
Music by Vincent Rubinetti
Download the music on Bandcamp: https://vincerubinetti.bandcamp.com/album/the-music-of-3blue1brown
Stream the music on Spotify: https://open.spotify.com/album/1dVyjwS8FBqXhRunaG5W5u