├── LICENSE ├── README.md ├── arch.png ├── augmentation.py ├── class.png ├── datasets.py ├── detection ├── convert-pretrain-to-detectron2.py ├── infomin_configs │ ├── Base-RCNN-FPN.yaml │ ├── R_50_FPN_1x.yaml │ └── R_50_FPN_1x_infomin.yaml ├── train_net.py └── voc_fpn_1fc │ ├── .DS_Store │ ├── Base-RCNN-FPN.yaml │ ├── pascal_voc_R_50_FPN_24k.yaml │ └── pascal_voc_R_50_FPN_24k_infomin.yaml ├── engine.py ├── evaluate.py ├── evaluate_cluster.py ├── evaluate_vitlinear.py ├── formula.png ├── imagenet_lmdb.py ├── lars.py ├── objective.py ├── requirements.txt ├── train.py ├── utils.py ├── vision_transformer.py └── widen_resnet.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/README.md -------------------------------------------------------------------------------- /arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/arch.png -------------------------------------------------------------------------------- /augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/augmentation.py -------------------------------------------------------------------------------- /class.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/class.png -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/datasets.py -------------------------------------------------------------------------------- /detection/convert-pretrain-to-detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/convert-pretrain-to-detectron2.py -------------------------------------------------------------------------------- /detection/infomin_configs/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/infomin_configs/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /detection/infomin_configs/R_50_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/infomin_configs/R_50_FPN_1x.yaml -------------------------------------------------------------------------------- /detection/infomin_configs/R_50_FPN_1x_infomin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/infomin_configs/R_50_FPN_1x_infomin.yaml -------------------------------------------------------------------------------- /detection/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/train_net.py -------------------------------------------------------------------------------- /detection/voc_fpn_1fc/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/voc_fpn_1fc/.DS_Store -------------------------------------------------------------------------------- /detection/voc_fpn_1fc/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/voc_fpn_1fc/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /detection/voc_fpn_1fc/pascal_voc_R_50_FPN_24k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/voc_fpn_1fc/pascal_voc_R_50_FPN_24k.yaml -------------------------------------------------------------------------------- /detection/voc_fpn_1fc/pascal_voc_R_50_FPN_24k_infomin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/detection/voc_fpn_1fc/pascal_voc_R_50_FPN_24k_infomin.yaml -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/engine.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/evaluate.py -------------------------------------------------------------------------------- /evaluate_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/evaluate_cluster.py -------------------------------------------------------------------------------- /evaluate_vitlinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/evaluate_vitlinear.py -------------------------------------------------------------------------------- /formula.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/formula.png -------------------------------------------------------------------------------- /imagenet_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/imagenet_lmdb.py -------------------------------------------------------------------------------- /lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/lars.py -------------------------------------------------------------------------------- /objective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/objective.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/utils.py -------------------------------------------------------------------------------- /vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/vision_transformer.py -------------------------------------------------------------------------------- /widen_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/TWIST/HEAD/widen_resnet.py --------------------------------------------------------------------------------