CoMatch

CoMatch copied to clipboard

CoMatch: Semi-supervised Learning with Contrastive Graph Regularization, ICCV 2021 (Salesforce Research).

This is a PyTorch implementation of the CoMatch paper [Blog]:

@inproceedings{CoMatch,
	title={Semi-supervised Learning with Contrastive Graph Regularization},
	author={Junnan Li and Caiming Xiong and Steven C.H. Hoi},
	booktitle={ICCV},
	year={2021}
}

Requirements:

• PyTorch ≥ 1.4
• pip install tensorboard_logger
• download and extract cifar-10 dataset into ./data/

To perform semi-supervised learning on CIFAR-10 with 4 labels per class, run:

python Train_CoMatch.py --n-labeled 40 --seed 1 

The results using different random seeds are:

ImageNet

For ImageNet experiments, see ./imagenet/