├── CBSA_finetune_vit ├── attention.py ├── datasets.py ├── main.py ├── requirements.txt └── vision_transformer.py ├── CBSA_image_classification ├── attn_layers.py ├── coding_rate.py ├── crate.py ├── crate_hook.py ├── data │ ├── dataset.py │ └── randomaug.py ├── finetune.py ├── main.py ├── models.py ├── plot.py ├── requirements.txt └── utils.py ├── CBSA_semantic_segmentation ├── requirements.txt ├── segm │ ├── __pycache__ │ │ ├── config.cpython-38.pyc │ │ ├── engine.cpython-38.pyc │ │ ├── metrics.cpython-38.pyc │ │ └── train.cpython-38.pyc │ ├── config.py │ ├── config.yml │ ├── data │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── ade20k.cpython-38.pyc │ │ │ ├── base.cpython-38.pyc │ │ │ ├── cityscapes.cpython-38.pyc │ │ │ ├── factory.cpython-38.pyc │ │ │ ├── imagenet.cpython-38.pyc │ │ │ ├── loader.cpython-38.pyc │ │ │ ├── pascal_context.cpython-38.pyc │ │ │ └── utils.cpython-38.pyc │ │ ├── ade20k.py │ │ ├── base.py │ │ ├── cityscapes.py │ │ ├── config │ │ │ ├── ade20k.py │ │ │ ├── ade20k.yml │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes.yml │ │ │ ├── pascal_context.py │ │ │ └── pascal_context.yml │ │ ├── factory.py │ │ ├── imagenet.py │ │ ├── loader.py │ │ ├── pascal_context.py │ │ └── utils.py │ ├── engine.py │ ├── eval │ │ ├── __pycache__ │ │ │ └── miou.cpython-38.pyc │ │ ├── accuracy.py │ │ └── miou.py │ ├── inference.py │ ├── metrics.py │ ├── model │ │ ├── __pycache__ │ │ │ ├── dec_blocks.cpython-38.pyc │ │ │ ├── decoder.cpython-38.pyc │ │ │ ├── enc_blocks.cpython-38.pyc │ │ │ ├── factory.cpython-38.pyc │ │ │ ├── segmenter.cpython-38.pyc │ │ │ ├── utils.cpython-38.pyc │ │ │ └── vit.cpython-38.pyc │ │ ├── dec_blocks.py │ │ ├── decoder.py │ │ ├── enc_blocks.py │ │ ├── factory.py │ │ ├── segmenter.py │ │ ├── utils.py │ │ └── vit.py │ ├── optim │ │ ├── __pycache__ │ │ │ ├── factory.cpython-38.pyc │ │ │ └── scheduler.cpython-38.pyc │ │ ├── factory.py │ │ └── scheduler.py │ ├── scripts │ │ ├── prepare_ade20k.py │ │ ├── prepare_cityscapes.py │ │ ├── prepare_pcontext.py │ │ └── show_attn_map.py │ ├── train.py │ └── utils │ │ ├── __pycache__ │ │ ├── distributed.cpython-38.pyc │ │ ├── logger.cpython-38.pyc │ │ └── torch.cpython-38.pyc │ │ ├── distributed.py │ │ ├── download.py │ │ ├── lines.py │ │ ├── logger.py │ │ ├── logs.py │ │ └── torch.py └── setup.py ├── LICENSE ├── README.md └── assets ├── attn_maps.png ├── mcr2.png ├── overview_1030.jpg ├── synthetic.png ├── vit_adapt_224.png ├── vit_adapt_384.png ├── vit_flops_base.png └── vit_flops_tiny.png /CBSA_finetune_vit/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_finetune_vit/attention.py -------------------------------------------------------------------------------- /CBSA_finetune_vit/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_finetune_vit/datasets.py -------------------------------------------------------------------------------- /CBSA_finetune_vit/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_finetune_vit/main.py -------------------------------------------------------------------------------- /CBSA_finetune_vit/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_finetune_vit/requirements.txt -------------------------------------------------------------------------------- /CBSA_finetune_vit/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_finetune_vit/vision_transformer.py -------------------------------------------------------------------------------- /CBSA_image_classification/attn_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/attn_layers.py -------------------------------------------------------------------------------- /CBSA_image_classification/coding_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/coding_rate.py -------------------------------------------------------------------------------- /CBSA_image_classification/crate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/crate.py -------------------------------------------------------------------------------- /CBSA_image_classification/crate_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/crate_hook.py -------------------------------------------------------------------------------- /CBSA_image_classification/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/data/dataset.py -------------------------------------------------------------------------------- /CBSA_image_classification/data/randomaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/data/randomaug.py -------------------------------------------------------------------------------- /CBSA_image_classification/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/finetune.py -------------------------------------------------------------------------------- /CBSA_image_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/main.py -------------------------------------------------------------------------------- /CBSA_image_classification/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/models.py -------------------------------------------------------------------------------- /CBSA_image_classification/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/plot.py -------------------------------------------------------------------------------- /CBSA_image_classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/requirements.txt -------------------------------------------------------------------------------- /CBSA_image_classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_image_classification/utils.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/requirements.txt -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/__pycache__/config.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/__pycache__/config.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/__pycache__/engine.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/__pycache__/engine.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/__pycache__/train.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/__pycache__/train.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/config.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/config.yml -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__init__.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/ade20k.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/ade20k.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/base.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/base.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/cityscapes.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/cityscapes.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/factory.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/factory.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/imagenet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/imagenet.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/loader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/loader.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/pascal_context.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/pascal_context.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/ade20k.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/base.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/cityscapes.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/ade20k.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/ade20k.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/ade20k.yml -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/cityscapes.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/cityscapes.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/cityscapes.yml -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/pascal_context.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/config/pascal_context.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/config/pascal_context.yml -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/factory.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/imagenet.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/loader.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/pascal_context.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/data/utils.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/engine.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/eval/__pycache__/miou.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/eval/__pycache__/miou.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/eval/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/eval/accuracy.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/eval/miou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/eval/miou.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/inference.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/metrics.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/dec_blocks.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/dec_blocks.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/decoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/decoder.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/enc_blocks.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/enc_blocks.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/factory.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/factory.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/segmenter.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/segmenter.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/__pycache__/vit.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/__pycache__/vit.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/dec_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/dec_blocks.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/decoder.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/enc_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/enc_blocks.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/factory.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/segmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/segmenter.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/utils.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/model/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/model/vit.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/optim/__pycache__/factory.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/optim/__pycache__/factory.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/optim/__pycache__/scheduler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/optim/__pycache__/scheduler.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/optim/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/optim/factory.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/optim/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/optim/scheduler.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/scripts/prepare_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/scripts/prepare_ade20k.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/scripts/prepare_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/scripts/prepare_cityscapes.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/scripts/prepare_pcontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/scripts/prepare_pcontext.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/scripts/show_attn_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/scripts/show_attn_map.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/train.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/__pycache__/distributed.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/__pycache__/distributed.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/__pycache__/torch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/__pycache__/torch.cpython-38.pyc -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/distributed.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/download.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/lines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/lines.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/logger.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/logs.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/segm/utils/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/segm/utils/torch.py -------------------------------------------------------------------------------- /CBSA_semantic_segmentation/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/CBSA_semantic_segmentation/setup.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/README.md -------------------------------------------------------------------------------- /assets/attn_maps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/attn_maps.png -------------------------------------------------------------------------------- /assets/mcr2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/mcr2.png -------------------------------------------------------------------------------- /assets/overview_1030.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/overview_1030.jpg -------------------------------------------------------------------------------- /assets/synthetic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/synthetic.png -------------------------------------------------------------------------------- /assets/vit_adapt_224.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/vit_adapt_224.png -------------------------------------------------------------------------------- /assets/vit_adapt_384.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/vit_adapt_384.png -------------------------------------------------------------------------------- /assets/vit_flops_base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/vit_flops_base.png -------------------------------------------------------------------------------- /assets/vit_flops_tiny.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QishuaiWen/CBSA/HEAD/assets/vit_flops_tiny.png --------------------------------------------------------------------------------