├── .gitignore ├── LICENSE ├── README.md ├── configs ├── default.yml ├── group_vit_gcc_redcap_30e.yml └── group_vit_gcc_yfcc_30e.yml ├── convert_dataset ├── convert_coco_object.py ├── convert_yfcc14m.py ├── create_subset.py └── process_redcaps.py ├── datasets ├── __init__.py ├── bpe_simple_vocab_16e6.txt.gz ├── builder.py ├── formatting.py ├── imagenet_template.py └── tokenizer.py ├── demo ├── demo_seg.py └── examples │ ├── coco.jpg │ ├── ctx.jpg │ └── voc.jpg ├── figs ├── github_arch.gif ├── github_coco.gif ├── github_ctx.gif └── github_voc.gif ├── main_group_vit.py ├── main_seg.py ├── models ├── __init__.py ├── builder.py ├── group_vit.py ├── misc.py ├── multi_label_contrastive.py ├── transformer.py └── utils.py ├── segmentation ├── configs │ └── _base_ │ │ ├── custom_import.py │ │ └── datasets │ │ ├── coco.py │ │ ├── pascal_context.py │ │ └── pascal_voc12.py ├── datasets │ ├── __init__.py │ ├── coco_object.py │ ├── pascal_context.py │ └── pascal_voc.py └── evaluation │ ├── __init__.py │ ├── builder.py │ ├── group_palette.txt │ └── group_vit_seg.py ├── setup.cfg ├── tools ├── dist_launch.sh └── dist_mn_launch.sh └── utils ├── __init__.py ├── checkpoint.py ├── config.py ├── logger.py ├── lr_scheduler.py ├── misc.py └── optimizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/README.md -------------------------------------------------------------------------------- /configs/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/configs/default.yml -------------------------------------------------------------------------------- /configs/group_vit_gcc_redcap_30e.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/configs/group_vit_gcc_redcap_30e.yml -------------------------------------------------------------------------------- /configs/group_vit_gcc_yfcc_30e.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/configs/group_vit_gcc_yfcc_30e.yml -------------------------------------------------------------------------------- /convert_dataset/convert_coco_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/convert_dataset/convert_coco_object.py -------------------------------------------------------------------------------- /convert_dataset/convert_yfcc14m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/convert_dataset/convert_yfcc14m.py -------------------------------------------------------------------------------- /convert_dataset/create_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/convert_dataset/create_subset.py -------------------------------------------------------------------------------- /convert_dataset/process_redcaps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/convert_dataset/process_redcaps.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/builder.py -------------------------------------------------------------------------------- /datasets/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/formatting.py -------------------------------------------------------------------------------- /datasets/imagenet_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/imagenet_template.py -------------------------------------------------------------------------------- /datasets/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/datasets/tokenizer.py -------------------------------------------------------------------------------- /demo/demo_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/demo/demo_seg.py -------------------------------------------------------------------------------- /demo/examples/coco.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/demo/examples/coco.jpg -------------------------------------------------------------------------------- /demo/examples/ctx.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/demo/examples/ctx.jpg -------------------------------------------------------------------------------- /demo/examples/voc.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/demo/examples/voc.jpg -------------------------------------------------------------------------------- /figs/github_arch.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/figs/github_arch.gif -------------------------------------------------------------------------------- /figs/github_coco.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/figs/github_coco.gif -------------------------------------------------------------------------------- /figs/github_ctx.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/figs/github_ctx.gif -------------------------------------------------------------------------------- /figs/github_voc.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/figs/github_voc.gif -------------------------------------------------------------------------------- /main_group_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/main_group_vit.py -------------------------------------------------------------------------------- /main_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/main_seg.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/builder.py -------------------------------------------------------------------------------- /models/group_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/group_vit.py -------------------------------------------------------------------------------- /models/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/misc.py -------------------------------------------------------------------------------- /models/multi_label_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/multi_label_contrastive.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/models/utils.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/custom_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/configs/_base_/custom_import.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/configs/_base_/datasets/coco.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/configs/_base_/datasets/pascal_context.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/configs/_base_/datasets/pascal_voc12.py -------------------------------------------------------------------------------- /segmentation/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/datasets/__init__.py -------------------------------------------------------------------------------- /segmentation/datasets/coco_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/datasets/coco_object.py -------------------------------------------------------------------------------- /segmentation/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/datasets/pascal_context.py -------------------------------------------------------------------------------- /segmentation/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/datasets/pascal_voc.py -------------------------------------------------------------------------------- /segmentation/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/evaluation/__init__.py -------------------------------------------------------------------------------- /segmentation/evaluation/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/evaluation/builder.py -------------------------------------------------------------------------------- /segmentation/evaluation/group_palette.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/evaluation/group_palette.txt -------------------------------------------------------------------------------- /segmentation/evaluation/group_vit_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/segmentation/evaluation/group_vit_seg.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/setup.cfg -------------------------------------------------------------------------------- /tools/dist_launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/tools/dist_launch.sh -------------------------------------------------------------------------------- /tools/dist_mn_launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/tools/dist_mn_launch.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/lr_scheduler.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/GroupViT/HEAD/utils/optimizer.py --------------------------------------------------------------------------------