├── .gitignore ├── DDP_byol_ccrop.py ├── DDP_linear.py ├── DDP_moco_ccrop.py ├── DDP_simclr_ccrop.py ├── DDP_simsiam_ccrop.py ├── LICENSE ├── README.md ├── configs ├── IN1K │ ├── mocov1_ccrop.py │ ├── mocov1_rcrop.py │ ├── mocov2_ccrop.py │ └── mocov2_rcrop.py ├── IN200 │ ├── mocov1_ccrop.py │ ├── mocov1_rcrop.py │ ├── mocov2_ccrop.py │ └── mocov2_rcrop.py ├── linear │ ├── IN1K_res50.py │ ├── IN200_res50.py │ ├── cifar100_res18.py │ ├── cifar10_res18.py │ ├── stl10_res18.py │ └── tiny200_res18.py └── small │ ├── cifar10 │ ├── byol_ccrop.py │ ├── byol_rcrop.py │ ├── moco_ccrop.py │ ├── moco_rcrop.py │ ├── moco_simccrop.py │ ├── simclr_ccrop.py │ ├── simclr_rcrop.py │ ├── simsiam_ccrop.py │ └── simsiam_rcrop.py │ ├── cifar100 │ ├── byol_ccrop.py │ ├── byol_rcrop.py │ ├── moco_ccrop.py │ ├── moco_rcrop.py │ ├── moco_simccrop.py │ ├── simclr_ccrop.py │ ├── simclr_rcrop.py │ ├── simsiam_ccrop.py │ └── simsiam_rcrop.py │ ├── stl10 │ ├── simclr_ccrop.py │ └── simclr_rcrop.py │ └── tiny200 │ ├── simclr_ccrop.py │ └── simclr_rcrop.py ├── datasets ├── __init__.py ├── build.py ├── cifar.py ├── imagenet.py ├── imagenet200.class ├── imagenet_subset.py ├── stl10.py ├── tiny200.py └── transforms │ ├── ContrastiveCrop.py │ ├── __init__.py │ ├── build.py │ ├── imagenet.py │ ├── misc.py │ └── small.py ├── figs └── motivation.png ├── losses ├── NT_Xent.py ├── NT_Xent_dist.py ├── __init__.py └── build.py ├── models ├── __init__.py ├── build.py ├── byol.py ├── moco.py ├── resnet.py └── simsiam.py ├── requirements.txt └── utils ├── __init__.py ├── cmds.txt ├── config.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/.gitignore -------------------------------------------------------------------------------- /DDP_byol_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/DDP_byol_ccrop.py -------------------------------------------------------------------------------- /DDP_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/DDP_linear.py -------------------------------------------------------------------------------- /DDP_moco_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/DDP_moco_ccrop.py -------------------------------------------------------------------------------- /DDP_simclr_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/DDP_simclr_ccrop.py -------------------------------------------------------------------------------- /DDP_simsiam_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/DDP_simsiam_ccrop.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/README.md -------------------------------------------------------------------------------- /configs/IN1K/mocov1_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN1K/mocov1_ccrop.py -------------------------------------------------------------------------------- /configs/IN1K/mocov1_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN1K/mocov1_rcrop.py -------------------------------------------------------------------------------- /configs/IN1K/mocov2_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN1K/mocov2_ccrop.py -------------------------------------------------------------------------------- /configs/IN1K/mocov2_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN1K/mocov2_rcrop.py -------------------------------------------------------------------------------- /configs/IN200/mocov1_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN200/mocov1_ccrop.py -------------------------------------------------------------------------------- /configs/IN200/mocov1_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN200/mocov1_rcrop.py -------------------------------------------------------------------------------- /configs/IN200/mocov2_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN200/mocov2_ccrop.py -------------------------------------------------------------------------------- /configs/IN200/mocov2_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/IN200/mocov2_rcrop.py -------------------------------------------------------------------------------- /configs/linear/IN1K_res50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/IN1K_res50.py -------------------------------------------------------------------------------- /configs/linear/IN200_res50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/IN200_res50.py -------------------------------------------------------------------------------- /configs/linear/cifar100_res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/cifar100_res18.py -------------------------------------------------------------------------------- /configs/linear/cifar10_res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/cifar10_res18.py -------------------------------------------------------------------------------- /configs/linear/stl10_res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/stl10_res18.py -------------------------------------------------------------------------------- /configs/linear/tiny200_res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/linear/tiny200_res18.py -------------------------------------------------------------------------------- /configs/small/cifar10/byol_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/byol_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/byol_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/byol_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/moco_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/moco_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/moco_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/moco_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/moco_simccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/moco_simccrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/simclr_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/simclr_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/simclr_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/simclr_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/simsiam_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/simsiam_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar10/simsiam_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar10/simsiam_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/byol_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/byol_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/byol_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/byol_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/moco_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/moco_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/moco_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/moco_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/moco_simccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/moco_simccrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/simclr_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/simclr_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/simclr_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/simclr_rcrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/simsiam_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/simsiam_ccrop.py -------------------------------------------------------------------------------- /configs/small/cifar100/simsiam_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/cifar100/simsiam_rcrop.py -------------------------------------------------------------------------------- /configs/small/stl10/simclr_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/stl10/simclr_ccrop.py -------------------------------------------------------------------------------- /configs/small/stl10/simclr_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/stl10/simclr_rcrop.py -------------------------------------------------------------------------------- /configs/small/tiny200/simclr_ccrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/tiny200/simclr_ccrop.py -------------------------------------------------------------------------------- /configs/small/tiny200/simclr_rcrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/configs/small/tiny200/simclr_rcrop.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/build.py -------------------------------------------------------------------------------- /datasets/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/cifar.py -------------------------------------------------------------------------------- /datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/imagenet.py -------------------------------------------------------------------------------- /datasets/imagenet200.class: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/imagenet200.class -------------------------------------------------------------------------------- /datasets/imagenet_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/imagenet_subset.py -------------------------------------------------------------------------------- /datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/stl10.py -------------------------------------------------------------------------------- /datasets/tiny200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/tiny200.py -------------------------------------------------------------------------------- /datasets/transforms/ContrastiveCrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/ContrastiveCrop.py -------------------------------------------------------------------------------- /datasets/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/__init__.py -------------------------------------------------------------------------------- /datasets/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/build.py -------------------------------------------------------------------------------- /datasets/transforms/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/imagenet.py -------------------------------------------------------------------------------- /datasets/transforms/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/misc.py -------------------------------------------------------------------------------- /datasets/transforms/small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/datasets/transforms/small.py -------------------------------------------------------------------------------- /figs/motivation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/figs/motivation.png -------------------------------------------------------------------------------- /losses/NT_Xent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/losses/NT_Xent.py -------------------------------------------------------------------------------- /losses/NT_Xent_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/losses/NT_Xent_dist.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/losses/build.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/build.py -------------------------------------------------------------------------------- /models/byol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/byol.py -------------------------------------------------------------------------------- /models/moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/moco.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/simsiam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/models/simsiam.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | addict 2 | yapf 3 | diffdist 4 | tensorboard 5 | -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/cmds.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/utils/cmds.txt -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xyupeng/ContrastiveCrop/HEAD/utils/util.py --------------------------------------------------------------------------------