├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── GETTING_STARTED.md ├── INSTALL.md ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── configs ├── ade20k-150-panoptic │ ├── maskformer_panoptic_R101_bs16_720k.yaml │ └── maskformer_panoptic_R50_bs16_720k.yaml ├── ade20k-150 │ ├── Base-ADE20K-150.yaml │ ├── maskformer_R101_bs16_160k.yaml │ ├── maskformer_R101c_bs16_160k.yaml │ ├── maskformer_R50_bs16_160k.yaml │ ├── per_pixel_baseline_R50_bs16_160k.yaml │ ├── per_pixel_baseline_plus_R50_bs16_160k.yaml │ └── swin │ │ ├── maskformer_swin_base_IN21k_384_bs16_160k_res640.yaml │ │ ├── maskformer_swin_large_IN21k_384_bs16_160k_res640.yaml │ │ ├── maskformer_swin_small_bs16_160k.yaml │ │ └── maskformer_swin_tiny_bs16_160k.yaml ├── ade20k-full-847 │ ├── Base-ADE20KFull-847.yaml │ ├── maskformer_R101_bs16_200k.yaml │ ├── maskformer_R101c_bs16_200k.yaml │ ├── maskformer_R50_bs16_200k.yaml │ ├── per_pixel_baseline_R50_bs16_200k.yaml │ └── per_pixel_baseline_plus_R50_bs16_200k.yaml ├── cityscapes-19 │ ├── Base-Cityscapes-19.yaml │ ├── maskformer_R101_bs16_90k.yaml │ └── maskformer_R101c_bs16_90k.yaml ├── coco-panoptic │ ├── Base-COCO-PanopticSegmentation.yaml │ ├── maskformer_panoptic_R101_bs64_554k.yaml │ ├── maskformer_panoptic_R50_bs64_554k.yaml │ └── swin │ │ ├── maskformer_panoptic_swin_base_IN21k_384_bs64_554k.yaml │ │ ├── maskformer_panoptic_swin_large_IN21k_384_bs64_554k.yaml │ │ ├── maskformer_panoptic_swin_small_bs64_554k.yaml │ │ └── maskformer_panoptic_swin_tiny_bs64_554k.yaml ├── coco-stuff-10k-171 │ ├── Base-COCOStuff10K-171.yaml │ ├── maskformer_R101_bs32_60k.yaml │ ├── maskformer_R101c_bs32_60k.yaml │ ├── maskformer_R50_bs32_60k.yaml │ ├── per_pixel_baseline_R50_bs32_60k.yaml │ └── per_pixel_baseline_plus_R50_bs32_60k.yaml └── mapillary-vistas-65 │ ├── Base-MapillaryVistas-65.yaml │ └── maskformer_R50_bs16_300k.yaml ├── datasets ├── README.md ├── ade20k_instance_catid_mapping.txt ├── ade20k_instance_imgCatIds.json ├── prepare_ade20k_full_sem_seg.py ├── prepare_ade20k_pan_seg.py ├── prepare_ade20k_sem_seg.py └── prepare_coco_stuff_10k_v1.0_sem_seg.py ├── demo ├── README.md ├── demo.py └── predictor.py ├── mask_former ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── detr_panoptic_dataset_mapper.py │ │ ├── mask_former_panoptic_dataset_mapper.py │ │ └── mask_former_semantic_dataset_mapper.py │ └── datasets │ │ ├── __init__.py │ │ ├── register_ade20k_full.py │ │ ├── register_ade20k_panoptic.py │ │ ├── register_coco_stuff_10k.py │ │ └── register_mapillary_vistas.py ├── mask_former_model.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ └── swin.py │ ├── criterion.py │ ├── heads │ │ ├── __init__.py │ │ ├── mask_former_head.py │ │ ├── per_pixel_baseline.py │ │ └── pixel_decoder.py │ ├── matcher.py │ └── transformer │ │ ├── __init__.py │ │ ├── position_encoding.py │ │ ├── transformer.py │ │ └── transformer_predictor.py ├── test_time_augmentation.py └── utils │ ├── __init__.py │ └── misc.py ├── requirements.txt ├── tools ├── README.md ├── analyze_model.py ├── convert-pretrained-swin-model-to-d2.py ├── convert-torchvision-to-d2.py └── evaluate_pq_for_semantic_segmentation.py └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/GETTING_STARTED.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/README.md -------------------------------------------------------------------------------- /configs/ade20k-150-panoptic/maskformer_panoptic_R101_bs16_720k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150-panoptic/maskformer_panoptic_R101_bs16_720k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150-panoptic/maskformer_panoptic_R50_bs16_720k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150-panoptic/maskformer_panoptic_R50_bs16_720k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/Base-ADE20K-150.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/Base-ADE20K-150.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/maskformer_R101_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/maskformer_R101_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/maskformer_R101c_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/maskformer_R101c_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/maskformer_R50_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/maskformer_R50_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/per_pixel_baseline_R50_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/per_pixel_baseline_R50_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/per_pixel_baseline_plus_R50_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/per_pixel_baseline_plus_R50_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/swin/maskformer_swin_base_IN21k_384_bs16_160k_res640.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/swin/maskformer_swin_base_IN21k_384_bs16_160k_res640.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/swin/maskformer_swin_large_IN21k_384_bs16_160k_res640.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/swin/maskformer_swin_large_IN21k_384_bs16_160k_res640.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/swin/maskformer_swin_small_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/swin/maskformer_swin_small_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-150/swin/maskformer_swin_tiny_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-150/swin/maskformer_swin_tiny_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/Base-ADE20KFull-847.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/Base-ADE20KFull-847.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/maskformer_R101_bs16_200k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/maskformer_R101_bs16_200k.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/maskformer_R101c_bs16_200k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/maskformer_R101c_bs16_200k.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/maskformer_R50_bs16_200k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/maskformer_R50_bs16_200k.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/per_pixel_baseline_R50_bs16_200k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/per_pixel_baseline_R50_bs16_200k.yaml -------------------------------------------------------------------------------- /configs/ade20k-full-847/per_pixel_baseline_plus_R50_bs16_200k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/ade20k-full-847/per_pixel_baseline_plus_R50_bs16_200k.yaml -------------------------------------------------------------------------------- /configs/cityscapes-19/Base-Cityscapes-19.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/cityscapes-19/Base-Cityscapes-19.yaml -------------------------------------------------------------------------------- /configs/cityscapes-19/maskformer_R101_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/cityscapes-19/maskformer_R101_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes-19/maskformer_R101c_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/cityscapes-19/maskformer_R101c_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/Base-COCO-PanopticSegmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/Base-COCO-PanopticSegmentation.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/maskformer_panoptic_R101_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/maskformer_panoptic_R101_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/maskformer_panoptic_R50_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/maskformer_panoptic_R50_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/swin/maskformer_panoptic_swin_base_IN21k_384_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/swin/maskformer_panoptic_swin_base_IN21k_384_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/swin/maskformer_panoptic_swin_large_IN21k_384_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/swin/maskformer_panoptic_swin_large_IN21k_384_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/swin/maskformer_panoptic_swin_small_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/swin/maskformer_panoptic_swin_small_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-panoptic/swin/maskformer_panoptic_swin_tiny_bs64_554k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-panoptic/swin/maskformer_panoptic_swin_tiny_bs64_554k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/Base-COCOStuff10K-171.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/Base-COCOStuff10K-171.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/maskformer_R101_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/maskformer_R101_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/maskformer_R101c_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/maskformer_R101c_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/maskformer_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/maskformer_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/per_pixel_baseline_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/per_pixel_baseline_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/coco-stuff-10k-171/per_pixel_baseline_plus_R50_bs32_60k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/coco-stuff-10k-171/per_pixel_baseline_plus_R50_bs32_60k.yaml -------------------------------------------------------------------------------- /configs/mapillary-vistas-65/Base-MapillaryVistas-65.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/mapillary-vistas-65/Base-MapillaryVistas-65.yaml -------------------------------------------------------------------------------- /configs/mapillary-vistas-65/maskformer_R50_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/configs/mapillary-vistas-65/maskformer_R50_bs16_300k.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/ade20k_instance_catid_mapping.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/ade20k_instance_catid_mapping.txt -------------------------------------------------------------------------------- /datasets/ade20k_instance_imgCatIds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/ade20k_instance_imgCatIds.json -------------------------------------------------------------------------------- /datasets/prepare_ade20k_full_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/prepare_ade20k_full_sem_seg.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_pan_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/prepare_ade20k_pan_seg.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/prepare_ade20k_sem_seg.py -------------------------------------------------------------------------------- /datasets/prepare_coco_stuff_10k_v1.0_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/datasets/prepare_coco_stuff_10k_v1.0_sem_seg.py -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/demo/predictor.py -------------------------------------------------------------------------------- /mask_former/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/__init__.py -------------------------------------------------------------------------------- /mask_former/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/config.py -------------------------------------------------------------------------------- /mask_former/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/__init__.py -------------------------------------------------------------------------------- /mask_former/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask_former/data/dataset_mappers/detr_panoptic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/dataset_mappers/detr_panoptic_dataset_mapper.py -------------------------------------------------------------------------------- /mask_former/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py -------------------------------------------------------------------------------- /mask_former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py -------------------------------------------------------------------------------- /mask_former/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/datasets/__init__.py -------------------------------------------------------------------------------- /mask_former/data/datasets/register_ade20k_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/datasets/register_ade20k_full.py -------------------------------------------------------------------------------- /mask_former/data/datasets/register_ade20k_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/datasets/register_ade20k_panoptic.py -------------------------------------------------------------------------------- /mask_former/data/datasets/register_coco_stuff_10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/datasets/register_coco_stuff_10k.py -------------------------------------------------------------------------------- /mask_former/data/datasets/register_mapillary_vistas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/data/datasets/register_mapillary_vistas.py -------------------------------------------------------------------------------- /mask_former/mask_former_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/mask_former_model.py -------------------------------------------------------------------------------- /mask_former/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/__init__.py -------------------------------------------------------------------------------- /mask_former/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask_former/modeling/backbone/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/backbone/swin.py -------------------------------------------------------------------------------- /mask_former/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/criterion.py -------------------------------------------------------------------------------- /mask_former/modeling/heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask_former/modeling/heads/mask_former_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/heads/mask_former_head.py -------------------------------------------------------------------------------- /mask_former/modeling/heads/per_pixel_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/heads/per_pixel_baseline.py -------------------------------------------------------------------------------- /mask_former/modeling/heads/pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/heads/pixel_decoder.py -------------------------------------------------------------------------------- /mask_former/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/matcher.py -------------------------------------------------------------------------------- /mask_former/modeling/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask_former/modeling/transformer/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/transformer/position_encoding.py -------------------------------------------------------------------------------- /mask_former/modeling/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/transformer/transformer.py -------------------------------------------------------------------------------- /mask_former/modeling/transformer/transformer_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/modeling/transformer/transformer_predictor.py -------------------------------------------------------------------------------- /mask_former/test_time_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/test_time_augmentation.py -------------------------------------------------------------------------------- /mask_former/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /mask_former/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/mask_former/utils/misc.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | cython 2 | scipy 3 | shapely 4 | timm 5 | h5py -------------------------------------------------------------------------------- /tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/tools/README.md -------------------------------------------------------------------------------- /tools/analyze_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/tools/analyze_model.py -------------------------------------------------------------------------------- /tools/convert-pretrained-swin-model-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/tools/convert-pretrained-swin-model-to-d2.py -------------------------------------------------------------------------------- /tools/convert-torchvision-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/tools/convert-torchvision-to-d2.py -------------------------------------------------------------------------------- /tools/evaluate_pq_for_semantic_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/tools/evaluate_pq_for_semantic_segmentation.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/MaskFormer/HEAD/train_net.py --------------------------------------------------------------------------------