In this tutorial, we will show you how to use LightlyTrain to train
a model on your own dataset for image classification.
Self-Supervised Learning (SSL) is reshaping computer vision, just like LLMs reshaped text. The newly launched LightlyTrain framework empowers AI teams—no PhD required—to easily train robust, unbiased foundation models on their own datasets. Let’s dive into how SSL with LightlyTrain beats traditional methods
Imagine training better computer vision models—without labeling a single image.
That’s exactly what LightlyTrain offers.
It brings self-supervised pretraining to your real-world pipelines,
using your unlabeled image or video data to kickstart model training.
We’ll walk through how to load the model, modify it for your dataset, preprocess the images, load the trained weights, and run predictions—including drawing labels on the image using OpenCV.
LightlyTrain page: https://www.lightly.ai/lightlytrain?utm_source=youtube&utm_medium=description&utm_campaign=eran
LightlyTrain Github : https://github.com/lightly-ai/lightly-train
LightlyTrain Docs: https://docs.lightly.ai/train/stable/index.html
Lightly Discord: https://discord.gg/xvNJW94
🔥 What You’ll Learn:
✅ Download and prepare the dataset
✅ How to Pre-train your custom dataset
✅ How to fine-tune your model with a new dataset / categories
✅ Test the model
Code for the tutorial : https://ko-fi.com/s/9be1ebf845
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 : [email protected]
🌐 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
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#EranFeit #Visiontransformers #LightlyTrain
~~~~~~~~~~~~~~ Chapters ~~~~~~~~~~~~~
00:00 Introduction and Demo
05:06 Installation
12:23 Build the model and Train on the first dataset
17:00 Fine-tune the model on a target dataset
28:31 Test the model
~~~~~~~~~~~~~~ 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