├── LICENSE.txt ├── README.md ├── configs ├── ade20k-135 │ ├── mask2former_R101c_alltask_bs32_60k.yaml │ ├── mask2former_R50c_alltask_bs32_60k.yaml │ └── mask2former_learn_prompt_bs32_16k.yaml ├── ade20k-150 │ ├── Base-ADE20K-150.yaml │ └── mask2former_R50_bs32_60k.yaml ├── coco-stuff-164k-156 │ ├── mask2former_R101c_alltask_bs32_60k.yaml │ ├── mask2former_R101c_alltask_bs32_60k_test_semseg.yaml │ ├── mask2former_R50_alltask_bs32_60k.yaml │ └── mask2former_learn_prompt_bs32_16k.yaml ├── coco-stuff-164k-171 │ ├── Base-COCOStuff164K-171.yaml │ ├── mask2former_R50_bs32_60k.yaml │ └── maskformer_R50_bs32_60k.yaml ├── voc-11k-15 │ ├── mask2former_R101c_bs16_20k.yaml │ ├── mask2former_R101c_bs16_20k_test_cross.yaml │ ├── mask2former_R50_bs16_20k.yaml │ └── mask2former_learn_prompt_bs16_10k.yaml └── voc-11k-20 │ ├── Base-VOC11K-20.yaml │ ├── mask2former_R50_bs16_20k.yaml │ └── maskformer_R50_bs16_20k.yaml ├── datasets ├── ade20k_instance_catid_mapping.txt ├── ade20k_instance_imgCatIds.json ├── prepare_ade20k_ins_seg_novel.py ├── prepare_ade20k_pan_seg_novel.py ├── prepare_ade20k_sem_seg_novel.py ├── prepare_coco_alldata.py ├── prepare_coco_stuff_164k_sem_seg.py └── prepare_voc_sem_seg.py ├── imgs └── framework.png ├── mask2former ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── augmentations.py │ ├── build.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── ade_all_task_dataset_mapper.py │ │ ├── coco_full_task_new_baseline_dataset_mapper.py │ │ ├── mask_former_binary_full_dataset_mapper.py │ │ ├── mask_former_binary_semantic_dataset_mapper.py │ │ ├── mask_former_instance_dataset_mapper.py │ │ ├── mask_former_panoptic_dataset_mapper.py │ │ ├── mask_former_semantic_dataset_mapper.py │ │ └── proposal_classification_dataset_mapper.py │ └── datasets │ │ ├── __init__.py │ │ ├── register_ade20k_all_data.py │ │ ├── register_coco_full.py │ │ ├── register_coco_panoptic.py │ │ ├── register_coco_stuff.py │ │ ├── register_voc_seg.py │ │ └── utils.py ├── evaluation │ ├── __init__.py │ ├── classification_evaluation.py │ ├── generalized_sem_seg_evaluation.py │ ├── instance_evaluation.py │ ├── instance_seg_evaluation.py │ ├── panoptic_seg_evaluation.py │ └── pseudo_sem_seg_evaluation.py ├── mask_former_model.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── clip_resnet.py │ │ └── swin.py │ ├── clip_adapter │ │ ├── __init__.py │ │ ├── adapter.py │ │ ├── clip.py │ │ └── text_prompt.py │ ├── criterion.py │ ├── heads │ │ ├── __init__.py │ │ ├── mask_former_head.py │ │ ├── mask_former_interaction_head.py │ │ ├── module.py │ │ ├── msdeformattn.py │ │ ├── ops │ │ │ ├── functions │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ │ └── ms_deform_attn_func.cpython-37.pyc │ │ │ │ └── ms_deform_attn_func.py │ │ │ ├── make.sh │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ │ └── ms_deform_attn.cpython-37.pyc │ │ │ │ └── ms_deform_attn.py │ │ │ ├── setup.py │ │ │ ├── src │ │ │ │ ├── cpu │ │ │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ │ │ └── ms_deform_attn_cpu.h │ │ │ │ ├── cuda │ │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ │ ├── ms_deform_attn_cuda.h │ │ │ │ │ └── ms_deform_im2col_cuda.cuh │ │ │ │ ├── ms_deform_attn.h │ │ │ │ └── vision.cpp │ │ │ └── test.py │ │ ├── per_pixel_baseline.py │ │ └── pixel_decoder.py │ ├── matcher.py │ └── transformer │ │ ├── __init__.py │ │ ├── mask2former_transformer_predictor.py │ │ ├── open_transformer_predictor.py │ │ ├── position_encoding.py │ │ ├── transformer.py │ │ └── transformer_predictor.py ├── open_vocabulary_model.py ├── proposal_classification.py ├── test_time_augmentation.py └── utils │ ├── __init__.py │ ├── events.py │ ├── misc.py │ ├── post_process_utils.py │ └── selective_search.py ├── requirements.txt ├── third_party └── CLIP │ ├── CLIP.png │ ├── LICENSE │ ├── MANIFEST.in │ ├── README.md │ ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py │ ├── model-card.md │ ├── requirements.txt │ ├── setup.py │ └── tests │ └── test_consistency.py ├── tools ├── convert-pretrained-clip-model-to-d2.py ├── convert-pretrained-swin-model-to-d2.py ├── convert-torchvision-to-d2.py ├── mask_cls_collect.py └── parse_name.py └── train_net.py /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/README.md -------------------------------------------------------------------------------- /configs/ade20k-135/mask2former_R101c_alltask_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/ade20k-135/mask2former_R101c_alltask_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/ade20k-135/mask2former_R50c_alltask_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/ade20k-135/mask2former_R50c_alltask_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/ade20k-135/mask2former_learn_prompt_bs32_16k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/ade20k-135/mask2former_learn_prompt_bs32_16k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/Base-ADE20K-150.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/ade20k-150/Base-ADE20K-150.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/mask2former_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/ade20k-150/mask2former_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-156/mask2former_R101c_alltask_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-156/mask2former_R101c_alltask_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-156/mask2former_R101c_alltask_bs32_60k_test_semseg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-156/mask2former_R101c_alltask_bs32_60k_test_semseg.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-156/mask2former_R50_alltask_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-156/mask2former_R50_alltask_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-156/mask2former_learn_prompt_bs32_16k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-156/mask2former_learn_prompt_bs32_16k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-171/Base-COCOStuff164K-171.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-171/Base-COCOStuff164K-171.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-171/mask2former_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-171/mask2former_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-164k-171/maskformer_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/coco-stuff-164k-171/maskformer_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/voc-11k-15/mask2former_R101c_bs16_20k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-15/mask2former_R101c_bs16_20k.yaml -------------------------------------------------------------------------------- /configs/voc-11k-15/mask2former_R101c_bs16_20k_test_cross.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-15/mask2former_R101c_bs16_20k_test_cross.yaml -------------------------------------------------------------------------------- /configs/voc-11k-15/mask2former_R50_bs16_20k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-15/mask2former_R50_bs16_20k.yaml -------------------------------------------------------------------------------- /configs/voc-11k-15/mask2former_learn_prompt_bs16_10k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-15/mask2former_learn_prompt_bs16_10k.yaml -------------------------------------------------------------------------------- /configs/voc-11k-20/Base-VOC11K-20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-20/Base-VOC11K-20.yaml -------------------------------------------------------------------------------- /configs/voc-11k-20/mask2former_R50_bs16_20k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-20/mask2former_R50_bs16_20k.yaml -------------------------------------------------------------------------------- /configs/voc-11k-20/maskformer_R50_bs16_20k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/configs/voc-11k-20/maskformer_R50_bs16_20k.yaml -------------------------------------------------------------------------------- /datasets/ade20k_instance_catid_mapping.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/ade20k_instance_catid_mapping.txt -------------------------------------------------------------------------------- /datasets/ade20k_instance_imgCatIds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/ade20k_instance_imgCatIds.json -------------------------------------------------------------------------------- /datasets/prepare_ade20k_ins_seg_novel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_ade20k_ins_seg_novel.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_pan_seg_novel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_ade20k_pan_seg_novel.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_sem_seg_novel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_ade20k_sem_seg_novel.py -------------------------------------------------------------------------------- /datasets/prepare_coco_alldata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_coco_alldata.py -------------------------------------------------------------------------------- /datasets/prepare_coco_stuff_164k_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_coco_stuff_164k_sem_seg.py -------------------------------------------------------------------------------- /datasets/prepare_voc_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/datasets/prepare_voc_sem_seg.py -------------------------------------------------------------------------------- /imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/imgs/framework.png -------------------------------------------------------------------------------- /mask2former/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/__init__.py -------------------------------------------------------------------------------- /mask2former/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/config.py -------------------------------------------------------------------------------- /mask2former/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/__init__.py -------------------------------------------------------------------------------- /mask2former/data/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/augmentations.py -------------------------------------------------------------------------------- /mask2former/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/build.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/__init__.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/ade_all_task_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/ade_all_task_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/coco_full_task_new_baseline_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/coco_full_task_new_baseline_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/mask_former_binary_full_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/mask_former_binary_full_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/mask_former_binary_semantic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/mask_former_binary_semantic_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/mask_former_instance_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/mask_former_instance_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/dataset_mappers/proposal_classification_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/dataset_mappers/proposal_classification_dataset_mapper.py -------------------------------------------------------------------------------- /mask2former/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/__init__.py -------------------------------------------------------------------------------- /mask2former/data/datasets/register_ade20k_all_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/register_ade20k_all_data.py -------------------------------------------------------------------------------- /mask2former/data/datasets/register_coco_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/register_coco_full.py -------------------------------------------------------------------------------- /mask2former/data/datasets/register_coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/register_coco_panoptic.py -------------------------------------------------------------------------------- /mask2former/data/datasets/register_coco_stuff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/register_coco_stuff.py -------------------------------------------------------------------------------- /mask2former/data/datasets/register_voc_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/register_voc_seg.py -------------------------------------------------------------------------------- /mask2former/data/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/data/datasets/utils.py -------------------------------------------------------------------------------- /mask2former/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/__init__.py -------------------------------------------------------------------------------- /mask2former/evaluation/classification_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/classification_evaluation.py -------------------------------------------------------------------------------- /mask2former/evaluation/generalized_sem_seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/generalized_sem_seg_evaluation.py -------------------------------------------------------------------------------- /mask2former/evaluation/instance_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/instance_evaluation.py -------------------------------------------------------------------------------- /mask2former/evaluation/instance_seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/instance_seg_evaluation.py -------------------------------------------------------------------------------- /mask2former/evaluation/panoptic_seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/panoptic_seg_evaluation.py -------------------------------------------------------------------------------- /mask2former/evaluation/pseudo_sem_seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/evaluation/pseudo_sem_seg_evaluation.py -------------------------------------------------------------------------------- /mask2former/mask_former_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/mask_former_model.py -------------------------------------------------------------------------------- /mask2former/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/__init__.py -------------------------------------------------------------------------------- /mask2former/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask2former/modeling/backbone/clip_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/backbone/clip_resnet.py -------------------------------------------------------------------------------- /mask2former/modeling/backbone/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/backbone/swin.py -------------------------------------------------------------------------------- /mask2former/modeling/clip_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/clip_adapter/__init__.py -------------------------------------------------------------------------------- /mask2former/modeling/clip_adapter/adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/clip_adapter/adapter.py -------------------------------------------------------------------------------- /mask2former/modeling/clip_adapter/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/clip_adapter/clip.py -------------------------------------------------------------------------------- /mask2former/modeling/clip_adapter/text_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/clip_adapter/text_prompt.py -------------------------------------------------------------------------------- /mask2former/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/criterion.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask2former/modeling/heads/mask_former_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/mask_former_head.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/mask_former_interaction_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/mask_former_interaction_head.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/module.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/msdeformattn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/msdeformattn.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/functions/__init__.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/functions/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/functions/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/functions/__pycache__/ms_deform_attn_func.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/functions/__pycache__/ms_deform_attn_func.cpython-37.pyc -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/make.sh -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/modules/__init__.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/modules/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/modules/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/modules/__pycache__/ms_deform_attn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/modules/__pycache__/ms_deform_attn.cpython-37.pyc -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/setup.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/src/vision.cpp -------------------------------------------------------------------------------- /mask2former/modeling/heads/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/ops/test.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/per_pixel_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/per_pixel_baseline.py -------------------------------------------------------------------------------- /mask2former/modeling/heads/pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/heads/pixel_decoder.py -------------------------------------------------------------------------------- /mask2former/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/matcher.py -------------------------------------------------------------------------------- /mask2former/modeling/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask2former/modeling/transformer/mask2former_transformer_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/transformer/mask2former_transformer_predictor.py -------------------------------------------------------------------------------- /mask2former/modeling/transformer/open_transformer_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/transformer/open_transformer_predictor.py -------------------------------------------------------------------------------- /mask2former/modeling/transformer/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/transformer/position_encoding.py -------------------------------------------------------------------------------- /mask2former/modeling/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/transformer/transformer.py -------------------------------------------------------------------------------- /mask2former/modeling/transformer/transformer_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/modeling/transformer/transformer_predictor.py -------------------------------------------------------------------------------- /mask2former/open_vocabulary_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/open_vocabulary_model.py -------------------------------------------------------------------------------- /mask2former/proposal_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/proposal_classification.py -------------------------------------------------------------------------------- /mask2former/test_time_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/test_time_augmentation.py -------------------------------------------------------------------------------- /mask2former/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/utils/__init__.py -------------------------------------------------------------------------------- /mask2former/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/utils/events.py -------------------------------------------------------------------------------- /mask2former/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/utils/misc.py -------------------------------------------------------------------------------- /mask2former/utils/post_process_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/utils/post_process_utils.py -------------------------------------------------------------------------------- /mask2former/utils/selective_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/mask2former/utils/selective_search.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/requirements.txt -------------------------------------------------------------------------------- /third_party/CLIP/CLIP.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/CLIP.png -------------------------------------------------------------------------------- /third_party/CLIP/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/LICENSE -------------------------------------------------------------------------------- /third_party/CLIP/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include clip/bpe_simple_vocab_16e6.txt.gz 2 | -------------------------------------------------------------------------------- /third_party/CLIP/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/README.md -------------------------------------------------------------------------------- /third_party/CLIP/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /third_party/CLIP/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /third_party/CLIP/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/clip/clip.py -------------------------------------------------------------------------------- /third_party/CLIP/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/clip/model.py -------------------------------------------------------------------------------- /third_party/CLIP/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /third_party/CLIP/model-card.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/model-card.md -------------------------------------------------------------------------------- /third_party/CLIP/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/requirements.txt -------------------------------------------------------------------------------- /third_party/CLIP/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/setup.py -------------------------------------------------------------------------------- /third_party/CLIP/tests/test_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/third_party/CLIP/tests/test_consistency.py -------------------------------------------------------------------------------- /tools/convert-pretrained-clip-model-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/tools/convert-pretrained-clip-model-to-d2.py -------------------------------------------------------------------------------- /tools/convert-pretrained-swin-model-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/tools/convert-pretrained-swin-model-to-d2.py -------------------------------------------------------------------------------- /tools/convert-torchvision-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/tools/convert-torchvision-to-d2.py -------------------------------------------------------------------------------- /tools/mask_cls_collect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/tools/mask_cls_collect.py -------------------------------------------------------------------------------- /tools/parse_name.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/tools/parse_name.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/FreeSeg/HEAD/train_net.py --------------------------------------------------------------------------------