learnopencv/SimSiam-Streamlining-SSL-with-Stop-Gradient-Mechanism

SimSiam: Streamlining SSL with Stop-Gradient Mechanism

This repository contains SimSiam (Simple Siamese) Neural Networks Pretraining implementation code on CIFAR-10 dataset and even the code for fine-tuning SimSiam's learned representations downstreamed for Classification task on the same CIFAR-10 dataset.

It is a part of the LearnOpenCV blog post - SimSiam: Streamlining SSL with Stop-Gradient Mechanism.

How to Execute?

Execute SimSiam_Pretraining_and_Fine_Tuning.ipynb notebook simply as it contains the PyTorch implementation code for pretraining SimSiam and also the code for fine-tuning SimSiam's learned representations downstreamed for classification on CIFAR-10 dataset.

AI Courses by OpenCV

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