├── .gitignore ├── LICENSE ├── README.md ├── contrast ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── rand_augment.py │ ├── transform.py │ ├── transform_coord.py │ └── zipreader.py ├── lars.py ├── logger.py ├── lr_scheduler.py ├── models │ ├── PixPro.py │ ├── __init__.py │ └── base.py ├── option.py ├── resnet.py └── util.py ├── demo ├── github_pixpro_pipeline.png └── github_teaser.png ├── main_linear.py ├── main_pretrain.py ├── publish_model.py ├── requirements.txt ├── tools ├── pixpro_base_r50_100ep.sh └── pixpro_base_r50_400ep.sh └── transfer └── detection ├── configs ├── Base-RCNN-C4-BN.yaml ├── Base-RCNN-FPN.yaml ├── COCO_R_50_C4_1x.yaml ├── COCO_R_50_C4_1x_PixPro.yaml ├── COCO_R_50_FPN_1x.yaml ├── COCO_R_50_FPN_1x_PixPro.yaml ├── Pascal_VOC_R_50_C4_24k.yaml └── Pascal_VOC_R_50_C4_24k_PixPro.yaml ├── convert_pretrain_to_d2.py ├── publish_model.py └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/README.md -------------------------------------------------------------------------------- /contrast/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /contrast/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/__init__.py -------------------------------------------------------------------------------- /contrast/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/dataset.py -------------------------------------------------------------------------------- /contrast/data/rand_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/rand_augment.py -------------------------------------------------------------------------------- /contrast/data/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/transform.py -------------------------------------------------------------------------------- /contrast/data/transform_coord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/transform_coord.py -------------------------------------------------------------------------------- /contrast/data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/data/zipreader.py -------------------------------------------------------------------------------- /contrast/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/lars.py -------------------------------------------------------------------------------- /contrast/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/logger.py -------------------------------------------------------------------------------- /contrast/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/lr_scheduler.py -------------------------------------------------------------------------------- /contrast/models/PixPro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/models/PixPro.py -------------------------------------------------------------------------------- /contrast/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/models/__init__.py -------------------------------------------------------------------------------- /contrast/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/models/base.py -------------------------------------------------------------------------------- /contrast/option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/option.py -------------------------------------------------------------------------------- /contrast/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/resnet.py -------------------------------------------------------------------------------- /contrast/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/contrast/util.py -------------------------------------------------------------------------------- /demo/github_pixpro_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/demo/github_pixpro_pipeline.png -------------------------------------------------------------------------------- /demo/github_teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/demo/github_teaser.png -------------------------------------------------------------------------------- /main_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/main_linear.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/publish_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/pixpro_base_r50_100ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/tools/pixpro_base_r50_100ep.sh -------------------------------------------------------------------------------- /tools/pixpro_base_r50_400ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/tools/pixpro_base_r50_400ep.sh -------------------------------------------------------------------------------- /transfer/detection/configs/Base-RCNN-C4-BN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/Base-RCNN-C4-BN.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/COCO_R_50_C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/COCO_R_50_C4_1x.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/COCO_R_50_C4_1x_PixPro.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/COCO_R_50_C4_1x_PixPro.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/COCO_R_50_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/COCO_R_50_FPN_1x.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/COCO_R_50_FPN_1x_PixPro.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/COCO_R_50_FPN_1x_PixPro.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/Pascal_VOC_R_50_C4_24k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/Pascal_VOC_R_50_C4_24k.yaml -------------------------------------------------------------------------------- /transfer/detection/configs/Pascal_VOC_R_50_C4_24k_PixPro.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/configs/Pascal_VOC_R_50_C4_24k_PixPro.yaml -------------------------------------------------------------------------------- /transfer/detection/convert_pretrain_to_d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/convert_pretrain_to_d2.py -------------------------------------------------------------------------------- /transfer/detection/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/publish_model.py -------------------------------------------------------------------------------- /transfer/detection/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zdaxie/PixPro/HEAD/transfer/detection/train_net.py --------------------------------------------------------------------------------