learnopencv/Contrastive-Learning-SimCLR-and-BYOL

Contrastive Learning : SimCLR and BYOL (With Code Example)

This repository contains BYOL (Bootstrap Your Own Latent) pretraining implementation code on CIFAR-10 dataset and also the code for fine-tuning BYOL's learned representations on STL-10 dataset.

It is part of the LearnOpenCV blog post - Contrastive Learning : SimCLR and BYOL (With Code Example).

How to Execute?

Execute the BYOL_implementation_in_Pytorch.ipynb notebook for pretraining the BYOL. And then execute the Linear_Feature_Eval_Raw.ipynb notebook for testing the fine-tuning purpose for downstream tasks.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.