├── .gitignore ├── LICENSE ├── README.md ├── assets ├── overview.png └── vis_map.png ├── classification ├── README.md ├── __init__.py ├── configs │ ├── defocus_mamba_base.yaml │ ├── defocus_mamba_large.yaml │ └── defocus_mamba_large_22k.yaml ├── data │ ├── CephDataset.py │ ├── __init__.py │ ├── build.py │ ├── cached_image_folder.py │ ├── data_simmim_ft.py │ ├── data_simmim_pt.py │ ├── imagenet.py │ ├── imagenet22k_dataset.py │ ├── map22kto1k.txt │ ├── samplers.py │ └── zipreader.py ├── dataset │ └── ImageNet1k │ │ └── val.txt ├── main.py ├── models │ ├── __init__.py │ ├── defocus_attention_network.py │ └── mamba_block.py ├── scripts │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ └── slurm_train.sh └── utils │ ├── config.py │ ├── init_env.py │ ├── logger.py │ ├── lr_scheduler.py │ ├── optimizer.py │ └── utils.py ├── detection ├── README.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── coco_detection.py │ │ │ └── coco_instance.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ └── dino-4scale_r50_8xb2-12e_coco.py │ │ └── schedules │ │ │ └── schedule_1x.py │ └── dino │ │ ├── defocus_mamba_base_12ep.py │ │ ├── defocus_mamba_base_36ep.py │ │ ├── defocus_mamba_large_12ep.py │ │ ├── defocus_mamba_large_36ep.py │ │ └── mamba_training_recipe.py ├── mmdet_custom │ ├── __init__.py │ └── models │ │ ├── __init__.py │ │ ├── mamba │ │ ├── __init__.py │ │ ├── defocus_mamba_backbone.py │ │ ├── mamba_block.py │ │ ├── mamba_config.py │ │ └── simple_fpn.py │ │ └── utils │ │ ├── __init__.py │ │ ├── drop.py │ │ └── embed.py └── scripts │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py └── tools ├── convert_22k_to_1k.py └── convert_cls_to_det.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/README.md -------------------------------------------------------------------------------- /assets/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/assets/overview.png -------------------------------------------------------------------------------- /assets/vis_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/assets/vis_map.png -------------------------------------------------------------------------------- /classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/README.md -------------------------------------------------------------------------------- /classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /classification/configs/defocus_mamba_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/configs/defocus_mamba_base.yaml -------------------------------------------------------------------------------- /classification/configs/defocus_mamba_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/configs/defocus_mamba_large.yaml -------------------------------------------------------------------------------- /classification/configs/defocus_mamba_large_22k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/configs/defocus_mamba_large_22k.yaml -------------------------------------------------------------------------------- /classification/data/CephDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/CephDataset.py -------------------------------------------------------------------------------- /classification/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/__init__.py -------------------------------------------------------------------------------- /classification/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/build.py -------------------------------------------------------------------------------- /classification/data/cached_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/cached_image_folder.py -------------------------------------------------------------------------------- /classification/data/data_simmim_ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/data_simmim_ft.py -------------------------------------------------------------------------------- /classification/data/data_simmim_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/data_simmim_pt.py -------------------------------------------------------------------------------- /classification/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/imagenet.py -------------------------------------------------------------------------------- /classification/data/imagenet22k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/imagenet22k_dataset.py -------------------------------------------------------------------------------- /classification/data/map22kto1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/map22kto1k.txt -------------------------------------------------------------------------------- /classification/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/samplers.py -------------------------------------------------------------------------------- /classification/data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/data/zipreader.py -------------------------------------------------------------------------------- /classification/dataset/ImageNet1k/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/dataset/ImageNet1k/val.txt -------------------------------------------------------------------------------- /classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/main.py -------------------------------------------------------------------------------- /classification/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/models/__init__.py -------------------------------------------------------------------------------- /classification/models/defocus_attention_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/models/defocus_attention_network.py -------------------------------------------------------------------------------- /classification/models/mamba_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/models/mamba_block.py -------------------------------------------------------------------------------- /classification/scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/scripts/dist_test.sh -------------------------------------------------------------------------------- /classification/scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/scripts/dist_train.sh -------------------------------------------------------------------------------- /classification/scripts/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/scripts/slurm_test.sh -------------------------------------------------------------------------------- /classification/scripts/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/scripts/slurm_train.sh -------------------------------------------------------------------------------- /classification/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/config.py -------------------------------------------------------------------------------- /classification/utils/init_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/init_env.py -------------------------------------------------------------------------------- /classification/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/logger.py -------------------------------------------------------------------------------- /classification/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/lr_scheduler.py -------------------------------------------------------------------------------- /classification/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/optimizer.py -------------------------------------------------------------------------------- /classification/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/classification/utils/utils.py -------------------------------------------------------------------------------- /detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/README.md -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/dino-4scale_r50_8xb2-12e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/_base_/models/dino-4scale_r50_8xb2-12e_coco.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/dino/defocus_mamba_base_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/dino/defocus_mamba_base_12ep.py -------------------------------------------------------------------------------- /detection/configs/dino/defocus_mamba_base_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/dino/defocus_mamba_base_36ep.py -------------------------------------------------------------------------------- /detection/configs/dino/defocus_mamba_large_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/dino/defocus_mamba_large_12ep.py -------------------------------------------------------------------------------- /detection/configs/dino/defocus_mamba_large_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/dino/defocus_mamba_large_36ep.py -------------------------------------------------------------------------------- /detection/configs/dino/mamba_training_recipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/configs/dino/mamba_training_recipe.py -------------------------------------------------------------------------------- /detection/mmdet_custom/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import * -------------------------------------------------------------------------------- /detection/mmdet_custom/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .mamba import * 2 | -------------------------------------------------------------------------------- /detection/mmdet_custom/models/mamba/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/mamba/__init__.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/mamba/defocus_mamba_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/mamba/defocus_mamba_backbone.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/mamba/mamba_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/mamba/mamba_block.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/mamba/mamba_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/mamba/mamba_config.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/mamba/simple_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/mamba/simple_fpn.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/utils/__init__.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/utils/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/utils/drop.py -------------------------------------------------------------------------------- /detection/mmdet_custom/models/utils/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/mmdet_custom/models/utils/embed.py -------------------------------------------------------------------------------- /detection/scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/dist_test.sh -------------------------------------------------------------------------------- /detection/scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/dist_train.sh -------------------------------------------------------------------------------- /detection/scripts/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/slurm_test.sh -------------------------------------------------------------------------------- /detection/scripts/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/slurm_train.sh -------------------------------------------------------------------------------- /detection/scripts/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/test.py -------------------------------------------------------------------------------- /detection/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/detection/scripts/train.py -------------------------------------------------------------------------------- /tools/convert_22k_to_1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/tools/convert_22k_to_1k.py -------------------------------------------------------------------------------- /tools/convert_cls_to_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/De-focus-Attention-Networks/HEAD/tools/convert_cls_to_det.py --------------------------------------------------------------------------------