Allen-Z-4230 / MoCo-CIFAR10

Adaptation of momentum contrast for CIFAR10 Dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Momentum Contrast

This repository is reproduced code of "Momentum Contrast for Unsupervised Visual Representation Learning"

The paper link is here

How to use

Run the train script.

python train.py

Visualizing the features with t-SNE

The below figure visualizes the extracted features of MNIST with t-SNE.

python test.py

image

About

Adaptation of momentum contrast for CIFAR10 Dataset

License:MIT License


Languages

Language:Jupyter Notebook 97.1%Language:Python 2.9%