├── README.md ├── classification ├── README.md ├── cifar10 │ ├── cropmix.py │ ├── main.py │ └── models │ │ ├── __pycache__ │ │ ├── densenet.cpython-37.pyc │ │ ├── mobilenetv2.cpython-37.pyc │ │ ├── preactresnet.cpython-37.pyc │ │ ├── resnet.cpython-37.pyc │ │ ├── resnext.cpython-37.pyc │ │ ├── stochasticdepth.cpython-37.pyc │ │ ├── swin.cpython-37.pyc │ │ ├── vit_small.cpython-37.pyc │ │ ├── wideresidual.cpython-37.pyc │ │ └── xception.cpython-37.pyc │ │ ├── densenet.py │ │ ├── preactresnet.py │ │ ├── resnet.py │ │ ├── resnext.py │ │ ├── vit.py │ │ ├── vit_small.py │ │ ├── wideresidual.py │ │ └── xception.py ├── cifar100 │ ├── cropmix.py │ ├── main.py │ └── models │ │ ├── __pycache__ │ │ ├── densenet.cpython-37.pyc │ │ ├── mobilenetv2.cpython-37.pyc │ │ ├── preactresnet.cpython-37.pyc │ │ ├── resnet.cpython-37.pyc │ │ ├── resnext.cpython-37.pyc │ │ ├── stochasticdepth.cpython-37.pyc │ │ ├── swin.cpython-37.pyc │ │ ├── vit_small.cpython-37.pyc │ │ ├── wideresidual.cpython-37.pyc │ │ └── xception.cpython-37.pyc │ │ ├── densenet.py │ │ ├── preactresnet.py │ │ ├── resnet.py │ │ ├── resnext.py │ │ ├── vit.py │ │ ├── vit_small.py │ │ ├── wideresidual.py │ │ └── xception.py └── imagenet │ ├── cropmix.py │ └── main.py ├── contrastive ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── main_lincls.py ├── main_moco.py └── moco │ ├── __init__.py │ ├── builder.py │ └── loader.py ├── cropmix.py ├── imgs ├── classification.png ├── contrastive.png ├── cropmix.png ├── mim.png ├── mim2.png └── visu.png └── mae ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── FINETUNE.md ├── LICENSE ├── PRETRAIN.md ├── README.md ├── command ├── cropmix.py ├── demo └── mae_visualize.ipynb ├── engine_finetune.py ├── engine_pretrain.py ├── main_finetune.py ├── main_linprobe.py ├── main_pretrain.py ├── models_mae.py ├── models_vit.py ├── result.png ├── submitit_finetune.py ├── submitit_linprobe.py ├── submitit_pretrain.py └── util ├── __pycache__ ├── lr_sched.cpython-37.pyc ├── misc.cpython-37.pyc └── pos_embed.cpython-37.pyc ├── crop.py ├── datasets.py ├── lars.py ├── lr_decay.py ├── lr_sched.py ├── misc.py └── pos_embed.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/README.md -------------------------------------------------------------------------------- /classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/README.md -------------------------------------------------------------------------------- /classification/cifar10/cropmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/cropmix.py -------------------------------------------------------------------------------- /classification/cifar10/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/main.py -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/densenet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/densenet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/mobilenetv2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/mobilenetv2.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/preactresnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/preactresnet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/resnext.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/resnext.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/stochasticdepth.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/stochasticdepth.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/swin.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/swin.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/vit_small.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/vit_small.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/wideresidual.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/wideresidual.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/__pycache__/xception.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/__pycache__/xception.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar10/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/densenet.py -------------------------------------------------------------------------------- /classification/cifar10/models/preactresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/preactresnet.py -------------------------------------------------------------------------------- /classification/cifar10/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/resnet.py -------------------------------------------------------------------------------- /classification/cifar10/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/resnext.py -------------------------------------------------------------------------------- /classification/cifar10/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/vit.py -------------------------------------------------------------------------------- /classification/cifar10/models/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/vit_small.py -------------------------------------------------------------------------------- /classification/cifar10/models/wideresidual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/wideresidual.py -------------------------------------------------------------------------------- /classification/cifar10/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar10/models/xception.py -------------------------------------------------------------------------------- /classification/cifar100/cropmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/cropmix.py -------------------------------------------------------------------------------- /classification/cifar100/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/main.py -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/densenet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/densenet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/mobilenetv2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/mobilenetv2.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/preactresnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/preactresnet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/resnext.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/resnext.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/stochasticdepth.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/stochasticdepth.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/swin.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/swin.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/vit_small.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/vit_small.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/wideresidual.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/wideresidual.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/__pycache__/xception.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/__pycache__/xception.cpython-37.pyc -------------------------------------------------------------------------------- /classification/cifar100/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/densenet.py -------------------------------------------------------------------------------- /classification/cifar100/models/preactresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/preactresnet.py -------------------------------------------------------------------------------- /classification/cifar100/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/resnet.py -------------------------------------------------------------------------------- /classification/cifar100/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/resnext.py -------------------------------------------------------------------------------- /classification/cifar100/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/vit.py -------------------------------------------------------------------------------- /classification/cifar100/models/vit_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/vit_small.py -------------------------------------------------------------------------------- /classification/cifar100/models/wideresidual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/wideresidual.py -------------------------------------------------------------------------------- /classification/cifar100/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/cifar100/models/xception.py -------------------------------------------------------------------------------- /classification/imagenet/cropmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/imagenet/cropmix.py -------------------------------------------------------------------------------- /classification/imagenet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/classification/imagenet/main.py -------------------------------------------------------------------------------- /contrastive/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /contrastive/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/CONTRIBUTING.md -------------------------------------------------------------------------------- /contrastive/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/LICENSE -------------------------------------------------------------------------------- /contrastive/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/README.md -------------------------------------------------------------------------------- /contrastive/main_lincls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/main_lincls.py -------------------------------------------------------------------------------- /contrastive/main_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/main_moco.py -------------------------------------------------------------------------------- /contrastive/moco/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /contrastive/moco/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/moco/builder.py -------------------------------------------------------------------------------- /contrastive/moco/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/contrastive/moco/loader.py -------------------------------------------------------------------------------- /cropmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/cropmix.py -------------------------------------------------------------------------------- /imgs/classification.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/classification.png -------------------------------------------------------------------------------- /imgs/contrastive.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/contrastive.png -------------------------------------------------------------------------------- /imgs/cropmix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/cropmix.png -------------------------------------------------------------------------------- /imgs/mim.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/mim.png -------------------------------------------------------------------------------- /imgs/mim2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/mim2.png -------------------------------------------------------------------------------- /imgs/visu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/imgs/visu.png -------------------------------------------------------------------------------- /mae/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /mae/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/CONTRIBUTING.md -------------------------------------------------------------------------------- /mae/FINETUNE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/FINETUNE.md -------------------------------------------------------------------------------- /mae/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/LICENSE -------------------------------------------------------------------------------- /mae/PRETRAIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/PRETRAIN.md -------------------------------------------------------------------------------- /mae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/README.md -------------------------------------------------------------------------------- /mae/command: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/command -------------------------------------------------------------------------------- /mae/cropmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/cropmix.py -------------------------------------------------------------------------------- /mae/demo/mae_visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/demo/mae_visualize.ipynb -------------------------------------------------------------------------------- /mae/engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/engine_finetune.py -------------------------------------------------------------------------------- /mae/engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/engine_pretrain.py -------------------------------------------------------------------------------- /mae/main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/main_finetune.py -------------------------------------------------------------------------------- /mae/main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/main_linprobe.py -------------------------------------------------------------------------------- /mae/main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/main_pretrain.py -------------------------------------------------------------------------------- /mae/models_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/models_mae.py -------------------------------------------------------------------------------- /mae/models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/models_vit.py -------------------------------------------------------------------------------- /mae/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/result.png -------------------------------------------------------------------------------- /mae/submitit_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/submitit_finetune.py -------------------------------------------------------------------------------- /mae/submitit_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/submitit_linprobe.py -------------------------------------------------------------------------------- /mae/submitit_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/submitit_pretrain.py -------------------------------------------------------------------------------- /mae/util/__pycache__/lr_sched.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/__pycache__/lr_sched.cpython-37.pyc -------------------------------------------------------------------------------- /mae/util/__pycache__/misc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/__pycache__/misc.cpython-37.pyc -------------------------------------------------------------------------------- /mae/util/__pycache__/pos_embed.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/__pycache__/pos_embed.cpython-37.pyc -------------------------------------------------------------------------------- /mae/util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/crop.py -------------------------------------------------------------------------------- /mae/util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/datasets.py -------------------------------------------------------------------------------- /mae/util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/lars.py -------------------------------------------------------------------------------- /mae/util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/lr_decay.py -------------------------------------------------------------------------------- /mae/util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/lr_sched.py -------------------------------------------------------------------------------- /mae/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/misc.py -------------------------------------------------------------------------------- /mae/util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunlinHan/CropMix/HEAD/mae/util/pos_embed.py --------------------------------------------------------------------------------