HobbitLong/PyContrast

PyTorch implementation of Contrastive Learning methods

Python

1,992

57 commits

updated Oct 4, 2023

See the code

README

PyContrast

This repo lists recent contrastive learning papers, and includes code for many of them.

Paper List

Find awesome contrastive learning paper.

PyTorch Code

For the reference implementation of SoTA methods, (e.g., InstDis, CMC, MoCo, ...), see pycontrast.

Pre-trained Models

A set of ImageNet unsupervised pre-training models are provided. Find them in model zoo.

Object Detection

Unsupervised pre-training models outperform supervised pre-training models on PASCAL VOC and COCO. Find in detection.

Significant stargazers

Urchade Zaratiana

232 followers · starred Jun 2021

Xuanchi Ren

310 followers · starred Mar 2021

Yang An

520 followers · starred Aug 2020

Bo-Kyeong Kim

24 followers · starred Dec 2020

Similar repos

HobbitLong/SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

3,449

same org

HobbitLong/CMC

[arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis

1,338

same org

youngtboy/Awesome-Self-Supervised-Vision-Pretrain

A paper list of self-supervised pretrain method

asheeshcric/awesome-contrastive-self-supervised-learning

A comprehensive list of awesome contrastive self-supervised learning papers.

1,310

facebookresearch/swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

2,093

lightly-ai/lightly

A python library for self-supervised learning on images.

3,821

wvangansbeke/Unsupervised-Classification

SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]

1,460

vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by…

1,580

HobbitLong/PyContrast

PyTorch implementation of Contrastive Learning methods

Python

1,992

57 commits

updated Oct 4, 2023

See the code

README

PyContrast

This repo lists recent contrastive learning papers, and includes code for many of them.

Paper List

Find awesome contrastive learning paper.

PyTorch Code

For the reference implementation of SoTA methods, (e.g., InstDis, CMC, MoCo, ...), see pycontrast.

Pre-trained Models

A set of ImageNet unsupervised pre-training models are provided. Find them in model zoo.

Object Detection

Unsupervised pre-training models outperform supervised pre-training models on PASCAL VOC and COCO. Find in detection.

Significant stargazers

Urchade Zaratiana

232 followers · starred Jun 2021

Xuanchi Ren

310 followers · starred Mar 2021

Yang An

520 followers · starred Aug 2020

Bo-Kyeong Kim

24 followers · starred Dec 2020

Similar repos

HobbitLong/SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

3,449

same org

HobbitLong/CMC

[arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis

1,338

same org

youngtboy/Awesome-Self-Supervised-Vision-Pretrain

A paper list of self-supervised pretrain method

asheeshcric/awesome-contrastive-self-supervised-learning

A comprehensive list of awesome contrastive self-supervised learning papers.

1,310

facebookresearch/swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

2,093

lightly-ai/lightly

A python library for self-supervised learning on images.

3,821

wvangansbeke/Unsupervised-Classification

SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]

1,460

vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by…

1,580