├── .github ├── attnmap.png ├── corresp.png ├── framework.png ├── global_semantics.png ├── ibot.png └── local_semantic_parts.png ├── .gitignore ├── INSTALL.md ├── LICENSE ├── README.md ├── analysis ├── README.md ├── attention_map │ ├── video_generation.py │ └── visualize_attention.py ├── backgrounds_challenge │ ├── README.md │ ├── challenge_eval.py │ ├── in9_classes.txt │ ├── in9_eval.py │ ├── in_to_in9.json │ └── tools │ │ ├── datasets.py │ │ ├── folder.py │ │ └── model_utils.py ├── combine_ckpt.py ├── correspondence │ └── visualize_correspondence.py ├── corruptions_surf_variations │ └── eval.py ├── eval_bg_challenge.sh ├── eval_corr_surf_vari.sh ├── eval_linear_part.sh ├── eval_natural_adv_examp.sh ├── eval_occlusion.sh ├── eval_shuffle.sh ├── extract_pattern │ └── extract_topk_cluster.py ├── imagenet_models │ ├── __init__.py │ ├── custom_modules.py │ ├── resnet.py │ └── vision_transformer.py ├── linear_part │ └── eval.py ├── natural_adv_examples │ ├── calibration_tools.py │ └── eval.py ├── occlusion │ ├── dino.py │ └── eval.py ├── visualize_attn_map.sh └── visualize_corresp.sh ├── evaluation ├── README.md ├── classification_layer_decay │ ├── __init__.py │ ├── engine_for_finetuning.py │ ├── extract_backbone_weights.py │ ├── modeling_finetune.py │ ├── optim_factory.py │ ├── run_class_finetuning.py │ └── utils.py ├── eval_cls.py ├── eval_copy_detection.py ├── eval_image_retrieval.py ├── eval_knn.py ├── eval_linear.py ├── eval_linear_multi.py ├── eval_logistic_regression.py ├── eval_video_segmentation.py ├── object_detection │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ └── coco_instance.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── cascade_mask_rcnn_swin_fpn.py │ │ │ │ └── cascade_mask_rcnn_vit_fpn.py │ │ │ └── schedules │ │ │ │ └── schedule_1x.py │ │ └── cascade_rcnn │ │ │ ├── swin_base_giou_4conv1f_coco_3x.py │ │ │ ├── swin_small_giou_4conv1f_coco_3x.py │ │ │ ├── vit_base_giou_4conv1f_coco_3x.py │ │ │ └── vit_small_giou_4conv1f_coco_3x.py │ ├── mmcv_custom │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── layer_decay_optimizer_constructor.py │ │ ├── register_backbone.py │ │ └── runner │ │ │ ├── __init__.py │ │ │ ├── checkpoint.py │ │ │ └── epoch_based_runner.py │ ├── test.py │ └── train.py ├── semantic_segmentation │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── ade20k.py │ │ │ │ ├── ade20k_640x640.py │ │ │ │ └── ade20k_fast.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ └── upernet.py │ │ │ └── schedules │ │ │ │ ├── schedule_160k.py │ │ │ │ └── schedule_320k.py │ │ ├── linear │ │ │ ├── vit_base_512_ade20k_160k.py │ │ │ ├── vit_base_640_ade20k_160k.py │ │ │ ├── vit_large_512_ade20k_160k.py │ │ │ ├── vit_large_640_ade20k_160k.py │ │ │ └── vit_small_512_ade20k_160k.py │ │ └── upernet │ │ │ ├── vit_base_512_ade20k_160k.py │ │ │ ├── vit_base_512_ade20k_160k_ms.py │ │ │ ├── vit_base_512_ade20k_160k_pt2ft.py │ │ │ ├── vit_base_640_ade20k_160k.py │ │ │ ├── vit_base_640_ade20k_160k_ms.py │ │ │ ├── vit_base_640_ade20k_160k_pt2ft.py │ │ │ ├── vit_large_512_ade20k_160k.py │ │ │ ├── vit_large_512_ade20k_160k_ms.py │ │ │ ├── vit_large_512_ade20k_160k_pt2ft.py │ │ │ ├── vit_large_640_ade20k_160k.py │ │ │ ├── vit_large_640_ade20k_160k_ms.py │ │ │ ├── vit_large_640_ade20k_160k_pt2ft.py │ │ │ ├── vit_small_512_ade20k_160k.py │ │ │ ├── vit_small_512_ade20k_160k_ms.py │ │ │ └── vit_small_512_ade20k_160k_pt2ft.py │ ├── mmcv_custom │ │ ├── __init__.py │ │ ├── apex_runner │ │ │ ├── __init__.py │ │ │ ├── apex_iter_based_runner.py │ │ │ ├── checkpoint.py │ │ │ └── optimizer.py │ │ ├── checkpoint.py │ │ ├── layer_decay_optimizer_constructor.py │ │ ├── register_backbone.py │ │ ├── resize_transform.py │ │ └── train_api.py │ ├── test.py │ └── train.py ├── semi_supervised │ └── eval_cls.py └── unsupervised │ └── unsup_cls.py ├── extract_backbone_weights.py ├── loader.py ├── main_ibot.py ├── models ├── __init__.py ├── head.py ├── swin_transformer.py └── vision_transformer.py ├── notebooks └── iBOT_demo.ipynb ├── run.sh └── utils.py /.github/attnmap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/attnmap.png -------------------------------------------------------------------------------- /.github/corresp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/corresp.png -------------------------------------------------------------------------------- /.github/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/framework.png -------------------------------------------------------------------------------- /.github/global_semantics.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/global_semantics.png -------------------------------------------------------------------------------- /.github/ibot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/ibot.png -------------------------------------------------------------------------------- /.github/local_semantic_parts.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.github/local_semantic_parts.png -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/.gitignore -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/README.md -------------------------------------------------------------------------------- /analysis/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/README.md -------------------------------------------------------------------------------- /analysis/attention_map/video_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/attention_map/video_generation.py -------------------------------------------------------------------------------- /analysis/attention_map/visualize_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/attention_map/visualize_attention.py -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/README.md -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/challenge_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/challenge_eval.py -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/in9_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/in9_classes.txt -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/in9_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/in9_eval.py -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/in_to_in9.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/in_to_in9.json -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/tools/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/tools/datasets.py -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/tools/folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/tools/folder.py -------------------------------------------------------------------------------- /analysis/backgrounds_challenge/tools/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/backgrounds_challenge/tools/model_utils.py -------------------------------------------------------------------------------- /analysis/combine_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/combine_ckpt.py -------------------------------------------------------------------------------- /analysis/correspondence/visualize_correspondence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/correspondence/visualize_correspondence.py -------------------------------------------------------------------------------- /analysis/corruptions_surf_variations/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/corruptions_surf_variations/eval.py -------------------------------------------------------------------------------- /analysis/eval_bg_challenge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_bg_challenge.sh -------------------------------------------------------------------------------- /analysis/eval_corr_surf_vari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_corr_surf_vari.sh -------------------------------------------------------------------------------- /analysis/eval_linear_part.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_linear_part.sh -------------------------------------------------------------------------------- /analysis/eval_natural_adv_examp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_natural_adv_examp.sh -------------------------------------------------------------------------------- /analysis/eval_occlusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_occlusion.sh -------------------------------------------------------------------------------- /analysis/eval_shuffle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/eval_shuffle.sh -------------------------------------------------------------------------------- /analysis/extract_pattern/extract_topk_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/extract_pattern/extract_topk_cluster.py -------------------------------------------------------------------------------- /analysis/imagenet_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/imagenet_models/__init__.py -------------------------------------------------------------------------------- /analysis/imagenet_models/custom_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/imagenet_models/custom_modules.py -------------------------------------------------------------------------------- /analysis/imagenet_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/imagenet_models/resnet.py -------------------------------------------------------------------------------- /analysis/imagenet_models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/imagenet_models/vision_transformer.py -------------------------------------------------------------------------------- /analysis/linear_part/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/linear_part/eval.py -------------------------------------------------------------------------------- /analysis/natural_adv_examples/calibration_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/natural_adv_examples/calibration_tools.py -------------------------------------------------------------------------------- /analysis/natural_adv_examples/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/natural_adv_examples/eval.py -------------------------------------------------------------------------------- /analysis/occlusion/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/occlusion/dino.py -------------------------------------------------------------------------------- /analysis/occlusion/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/occlusion/eval.py -------------------------------------------------------------------------------- /analysis/visualize_attn_map.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/visualize_attn_map.sh -------------------------------------------------------------------------------- /analysis/visualize_corresp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/analysis/visualize_corresp.sh -------------------------------------------------------------------------------- /evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/README.md -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/engine_for_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/engine_for_finetuning.py -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/extract_backbone_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/extract_backbone_weights.py -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/modeling_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/modeling_finetune.py -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/optim_factory.py -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/run_class_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/run_class_finetuning.py -------------------------------------------------------------------------------- /evaluation/classification_layer_decay/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/classification_layer_decay/utils.py -------------------------------------------------------------------------------- /evaluation/eval_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_cls.py -------------------------------------------------------------------------------- /evaluation/eval_copy_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_copy_detection.py -------------------------------------------------------------------------------- /evaluation/eval_image_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_image_retrieval.py -------------------------------------------------------------------------------- /evaluation/eval_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_knn.py -------------------------------------------------------------------------------- /evaluation/eval_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_linear.py -------------------------------------------------------------------------------- /evaluation/eval_linear_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_linear_multi.py -------------------------------------------------------------------------------- /evaluation/eval_logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_logistic_regression.py -------------------------------------------------------------------------------- /evaluation/eval_video_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/eval_video_segmentation.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/_base_/models/cascade_mask_rcnn_swin_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/_base_/models/cascade_mask_rcnn_swin_fpn.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/_base_/models/cascade_mask_rcnn_vit_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/_base_/models/cascade_mask_rcnn_vit_fpn.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/cascade_rcnn/swin_base_giou_4conv1f_coco_3x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/cascade_rcnn/swin_base_giou_4conv1f_coco_3x.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/cascade_rcnn/swin_small_giou_4conv1f_coco_3x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/cascade_rcnn/swin_small_giou_4conv1f_coco_3x.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/cascade_rcnn/vit_base_giou_4conv1f_coco_3x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/cascade_rcnn/vit_base_giou_4conv1f_coco_3x.py -------------------------------------------------------------------------------- /evaluation/object_detection/configs/cascade_rcnn/vit_small_giou_4conv1f_coco_3x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/configs/cascade_rcnn/vit_small_giou_4conv1f_coco_3x.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/register_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/register_backbone.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/runner/__init__.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/runner/checkpoint.py -------------------------------------------------------------------------------- /evaluation/object_detection/mmcv_custom/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/mmcv_custom/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /evaluation/object_detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/test.py -------------------------------------------------------------------------------- /evaluation/object_detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/object_detection/train.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/datasets/ade20k_640x640.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/datasets/ade20k_640x640.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/datasets/ade20k_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/datasets/ade20k_fast.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/models/upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/models/upernet.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/linear/vit_base_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/linear/vit_base_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/linear/vit_base_640_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/linear/vit_base_640_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/linear/vit_large_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/linear/vit_large_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/linear/vit_large_640_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/linear/vit_large_640_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/linear/vit_small_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/linear/vit_small_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k_ms.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k_pt2ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_512_ade20k_160k_pt2ft.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k_ms.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k_pt2ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_base_640_ade20k_160k_pt2ft.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k_ms.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k_pt2ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_512_ade20k_160k_pt2ft.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k_ms.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k_pt2ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_large_640_ade20k_160k_pt2ft.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k_ms.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k_pt2ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/configs/upernet/vit_small_512_ade20k_160k_pt2ft.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/apex_runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/apex_runner/__init__.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/apex_runner/apex_iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/apex_runner/apex_iter_based_runner.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/apex_runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/apex_runner/checkpoint.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/apex_runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/apex_runner/optimizer.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/register_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/register_backbone.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/resize_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/resize_transform.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/mmcv_custom/train_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/mmcv_custom/train_api.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/test.py -------------------------------------------------------------------------------- /evaluation/semantic_segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semantic_segmentation/train.py -------------------------------------------------------------------------------- /evaluation/semi_supervised/eval_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/semi_supervised/eval_cls.py -------------------------------------------------------------------------------- /evaluation/unsupervised/unsup_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/evaluation/unsupervised/unsup_cls.py -------------------------------------------------------------------------------- /extract_backbone_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/extract_backbone_weights.py -------------------------------------------------------------------------------- /loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/loader.py -------------------------------------------------------------------------------- /main_ibot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/main_ibot.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/models/head.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/models/vision_transformer.py -------------------------------------------------------------------------------- /notebooks/iBOT_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/notebooks/iBOT_demo.ipynb -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/run.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/ibot/HEAD/utils.py --------------------------------------------------------------------------------