├── .gitignore ├── LICENSE ├── README.md ├── assets ├── R50.png ├── generalization.png ├── logo.png ├── memory_fusion.png ├── performance.png ├── sota.png ├── stable_maskdino.png ├── stable_matching.png └── swinl.png ├── projects └── stabledino │ ├── README.md │ ├── assets │ └── dino_arch.png │ ├── configs │ ├── data │ │ ├── coco_instance_seg.py │ │ └── coco_panoptic_seg.py │ ├── models │ │ ├── dino_focalnet.py │ │ ├── dino_r50.py │ │ ├── dino_swin_base_384.py │ │ ├── dino_swin_large_224.py │ │ ├── dino_swin_large_384.py │ │ ├── dino_swin_small_224.py │ │ ├── dino_swin_tiny_224.py │ │ ├── dino_vitdet.py │ │ ├── stabledino_r50.py │ │ └── stablegdino_swin_large_384.py │ ├── stabledino_r50_4scale_12ep.py │ ├── stabledino_swinL_4scale_12ep.py │ ├── timm_example.py │ └── torchvision_example.py │ ├── data │ ├── __init__.py │ └── dataset_mappers │ │ ├── __init__.py │ │ └── coco_instance_lsj_aug_dataset_mapper.py │ ├── modeling │ ├── __init__.py │ ├── dino.py │ ├── dino_transformer.py │ ├── dn_criterion.py │ ├── matcher.py │ └── two_stage_criterion.py │ └── utils │ ├── __init__.py │ ├── box_ops.py │ ├── misc.py │ ├── utils.py │ └── utils_nms.py └── tools └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/README.md -------------------------------------------------------------------------------- /assets/R50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/R50.png -------------------------------------------------------------------------------- /assets/generalization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/generalization.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/logo.png -------------------------------------------------------------------------------- /assets/memory_fusion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/memory_fusion.png -------------------------------------------------------------------------------- /assets/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/performance.png -------------------------------------------------------------------------------- /assets/sota.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/sota.png -------------------------------------------------------------------------------- /assets/stable_maskdino.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/stable_maskdino.png -------------------------------------------------------------------------------- /assets/stable_matching.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/stable_matching.png -------------------------------------------------------------------------------- /assets/swinl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/assets/swinl.png -------------------------------------------------------------------------------- /projects/stabledino/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/README.md -------------------------------------------------------------------------------- /projects/stabledino/assets/dino_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/assets/dino_arch.png -------------------------------------------------------------------------------- /projects/stabledino/configs/data/coco_instance_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/data/coco_instance_seg.py -------------------------------------------------------------------------------- /projects/stabledino/configs/data/coco_panoptic_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/data/coco_panoptic_seg.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_focalnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_focalnet.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_r50.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_swin_base_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_swin_base_384.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_swin_large_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_swin_large_224.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_swin_large_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_swin_large_384.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_swin_small_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_swin_small_224.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_swin_tiny_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_swin_tiny_224.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/dino_vitdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/dino_vitdet.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/stabledino_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/stabledino_r50.py -------------------------------------------------------------------------------- /projects/stabledino/configs/models/stablegdino_swin_large_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/models/stablegdino_swin_large_384.py -------------------------------------------------------------------------------- /projects/stabledino/configs/stabledino_r50_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/stabledino_r50_4scale_12ep.py -------------------------------------------------------------------------------- /projects/stabledino/configs/stabledino_swinL_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/stabledino_swinL_4scale_12ep.py -------------------------------------------------------------------------------- /projects/stabledino/configs/timm_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/timm_example.py -------------------------------------------------------------------------------- /projects/stabledino/configs/torchvision_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/configs/torchvision_example.py -------------------------------------------------------------------------------- /projects/stabledino/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/data/__init__.py -------------------------------------------------------------------------------- /projects/stabledino/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /projects/stabledino/data/dataset_mappers/coco_instance_lsj_aug_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/data/dataset_mappers/coco_instance_lsj_aug_dataset_mapper.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/__init__.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/dino.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/dino_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/dino_transformer.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/matcher.py -------------------------------------------------------------------------------- /projects/stabledino/modeling/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/modeling/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/stabledino/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | # import misc -------------------------------------------------------------------------------- /projects/stabledino/utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/utils/box_ops.py -------------------------------------------------------------------------------- /projects/stabledino/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/utils/misc.py -------------------------------------------------------------------------------- /projects/stabledino/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/utils/utils.py -------------------------------------------------------------------------------- /projects/stabledino/utils/utils_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/projects/stabledino/utils/utils_nms.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IDEA-Research/Stable-DINO/HEAD/tools/train_net.py --------------------------------------------------------------------------------