MOCO-tf2

This project implements constrastive learning algorithm introduced in Momentum Contrast for Unsupervised Visual Representation Learning with tensorflow 2.0

download dataset

The project trains ResNet50 with constrastive learning on imagenet resized version. Download the dataset with

python3 download_dataset.py

Train the model with the following command

python3 train.py

Name		Name	Last commit message	Last commit date
Latest commit History 53 Commits
AffineLayer.py		AffineLayer.py
README.md		README.md
SKL.py		SKL.py
download_dataset.py		download_dataset.py
models.py		models.py
save_model.py		save_model.py
test.py		test.py
train.py		train.py