├── .gitignore ├── GETTING_STARTED.md ├── INSTALL.md ├── LICENSE ├── README.md ├── colab └── oneformer_colab.ipynb ├── configs ├── ade20k │ ├── Base-ADE20K-UnifiedSegmentation.yaml │ ├── convnext │ │ ├── oneformer_convnext_large_bs16_160k.yaml │ │ └── oneformer_convnext_xlarge_bs16_160k.yaml │ ├── dinat │ │ ├── coco_pretrain_oneformer_dinat_large_bs16_160k_1280x1280.yaml │ │ ├── oneformer_dinat_large_bs16_160k.yaml │ │ ├── oneformer_dinat_large_bs16_160k_1280x1280.yaml │ │ └── oneformer_dinat_large_bs16_160k_896x896.yaml │ ├── oneformer_R50_bs16_160k.yaml │ └── swin │ │ ├── oneformer_swin_large_bs16_160k.yaml │ │ ├── oneformer_swin_large_bs16_160k_1280x1280.yaml │ │ ├── oneformer_swin_large_bs16_160k_896x896.yaml │ │ └── oneformer_swin_tiny_bs16_160k.yaml ├── cityscapes │ ├── Base-Cityscapes-UnifiedSegmentation.yaml │ ├── convnext │ │ ├── mapillary_pretrain_oneformer_convnext_large_bs16_90k.yaml │ │ ├── mapillary_pretrain_oneformer_convnext_xlarge_bs16_90k.yaml │ │ ├── oneformer_convnext_large_bs16_90k.yaml │ │ └── oneformer_convnext_xlarge_bs16_90k.yaml │ ├── dinat │ │ └── oneformer_dinat_large_bs16_90k.yaml │ ├── oneformer_R50_bs16_90k.yaml │ └── swin │ │ └── oneformer_swin_large_bs16_90k.yaml ├── coco │ ├── Base-COCO-UnifiedSegmentation.yaml │ ├── dinat │ │ └── oneformer_dinat_large_bs16_100ep.yaml │ ├── oneformer_R50_bs16_50ep.yaml │ └── swin │ │ ├── oneformer_swin_large_bs16_100ep.yaml │ │ └── oneformer_swin_tiny_bs16_50ep.yaml └── mapillary_vistas │ ├── Base-Mapillary-UnifiedSegmentation.yaml │ ├── convnext │ ├── cityscapes_pretrain_oneformer_convnext_large_bs16_300k.yaml │ ├── cityscapes_pretrain_oneformer_convnext_xlarge_bs16_300k.yaml │ └── oneformer_convnext_large_bs16_300k.yaml │ ├── dinat │ └── oneformer_dinat_large_bs16_300k.yaml │ ├── oneformer_R50_bs16_300k.yaml │ └── swin │ └── oneformer_swin_large_bs16_300k.yaml ├── datasets ├── README.md ├── ade20k_instance_catid_mapping.txt ├── ade20k_instance_imgCatIds.json ├── custom_datasets │ ├── README.md │ ├── instance_coco_custom_dataset_mapper.py │ ├── instance_oneformer_custom_dataset_mapper.py │ └── semantic_oneformer_custom_dataset_mapper.py ├── fg_ids.py ├── panoptic2detection_coco_format.py ├── panoptic_coco_categories.json ├── prepare_ade20k_ins_seg.py ├── prepare_ade20k_pan_seg.py ├── prepare_ade20k_sem_seg.py └── prepare_coco_semantic_annos_from_panoptic_annos.py ├── demo ├── README.md ├── colormap.py ├── defaults.py ├── demo.py ├── predictor.py └── visualizer.py ├── images ├── oneformer.svg ├── plots.svg ├── teaser.png └── teaser.svg ├── oneformer ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── build.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── coco_unified_new_baseline_dataset_mapper.py │ │ ├── dataset_mapper.py │ │ └── oneformer_unified_dataset_mapper.py │ ├── datasets │ │ ├── __init__.py │ │ ├── register_ade20k_instance.py │ │ ├── register_ade20k_panoptic.py │ │ ├── register_cityscapes_panoptic.py │ │ ├── register_coco_panoptic2instance.py │ │ ├── register_coco_panoptic_annos_semseg.py │ │ ├── register_mapillary_vistas.py │ │ └── register_mapillary_vistas_panoptic.py │ └── tokenizer.py ├── datasetmapper_tta.py ├── evaluation │ ├── __init__.py │ ├── cityscapes_evaluation.py │ ├── coco_evaluator.py │ ├── detection_coco_evaluator.py │ ├── evaluator.py │ └── instance_evaluation.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── convnext.py │ │ ├── dinat.py │ │ └── swin.py │ ├── criterion.py │ ├── matcher.py │ ├── meta_arch │ │ ├── __init__.py │ │ └── oneformer_head.py │ ├── pixel_decoder │ │ ├── __init__.py │ │ ├── fpn.py │ │ ├── msdeformattn.py │ │ └── ops │ │ │ ├── functions │ │ │ ├── __init__.py │ │ │ └── ms_deform_attn_func.py │ │ │ ├── make.sh │ │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── 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 │ └── transformer_decoder │ │ ├── __init__.py │ │ ├── oneformer_transformer_decoder.py │ │ ├── position_encoding.py │ │ ├── text_transformer.py │ │ └── transformer.py ├── oneformer_model.py ├── test_time_augmentation.py └── utils │ ├── __init__.py │ ├── box_ops.py │ ├── events.py │ ├── misc.py │ └── pos_embed.py ├── requirements.txt ├── tools ├── README.md ├── analyze_model.py ├── calc_throughput.py ├── convert-pretrained-model-to-d2.py ├── convert-pretrained-nat-model-to-d2.py ├── convert-torchvision-to-d2.py ├── setup_detectron2.py └── trainers │ ├── trainer.py │ └── trainer_base.py └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/GETTING_STARTED.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/README.md -------------------------------------------------------------------------------- /colab/oneformer_colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/colab/oneformer_colab.ipynb -------------------------------------------------------------------------------- /configs/ade20k/Base-ADE20K-UnifiedSegmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/Base-ADE20K-UnifiedSegmentation.yaml -------------------------------------------------------------------------------- /configs/ade20k/convnext/oneformer_convnext_large_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/convnext/oneformer_convnext_large_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k/convnext/oneformer_convnext_xlarge_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/convnext/oneformer_convnext_xlarge_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k/dinat/coco_pretrain_oneformer_dinat_large_bs16_160k_1280x1280.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/dinat/coco_pretrain_oneformer_dinat_large_bs16_160k_1280x1280.yaml -------------------------------------------------------------------------------- /configs/ade20k/dinat/oneformer_dinat_large_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/dinat/oneformer_dinat_large_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k/dinat/oneformer_dinat_large_bs16_160k_1280x1280.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/dinat/oneformer_dinat_large_bs16_160k_1280x1280.yaml -------------------------------------------------------------------------------- /configs/ade20k/dinat/oneformer_dinat_large_bs16_160k_896x896.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/dinat/oneformer_dinat_large_bs16_160k_896x896.yaml -------------------------------------------------------------------------------- /configs/ade20k/oneformer_R50_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/oneformer_R50_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k/swin/oneformer_swin_large_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/swin/oneformer_swin_large_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/ade20k/swin/oneformer_swin_large_bs16_160k_1280x1280.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/swin/oneformer_swin_large_bs16_160k_1280x1280.yaml -------------------------------------------------------------------------------- /configs/ade20k/swin/oneformer_swin_large_bs16_160k_896x896.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/swin/oneformer_swin_large_bs16_160k_896x896.yaml -------------------------------------------------------------------------------- /configs/ade20k/swin/oneformer_swin_tiny_bs16_160k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/ade20k/swin/oneformer_swin_tiny_bs16_160k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/Base-Cityscapes-UnifiedSegmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/Base-Cityscapes-UnifiedSegmentation.yaml -------------------------------------------------------------------------------- /configs/cityscapes/convnext/mapillary_pretrain_oneformer_convnext_large_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/convnext/mapillary_pretrain_oneformer_convnext_large_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/convnext/mapillary_pretrain_oneformer_convnext_xlarge_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/convnext/mapillary_pretrain_oneformer_convnext_xlarge_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/convnext/oneformer_convnext_large_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/convnext/oneformer_convnext_large_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/convnext/oneformer_convnext_xlarge_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/convnext/oneformer_convnext_xlarge_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/dinat/oneformer_dinat_large_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/dinat/oneformer_dinat_large_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/oneformer_R50_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/oneformer_R50_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/cityscapes/swin/oneformer_swin_large_bs16_90k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/cityscapes/swin/oneformer_swin_large_bs16_90k.yaml -------------------------------------------------------------------------------- /configs/coco/Base-COCO-UnifiedSegmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/coco/Base-COCO-UnifiedSegmentation.yaml -------------------------------------------------------------------------------- /configs/coco/dinat/oneformer_dinat_large_bs16_100ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/coco/dinat/oneformer_dinat_large_bs16_100ep.yaml -------------------------------------------------------------------------------- /configs/coco/oneformer_R50_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/coco/oneformer_R50_bs16_50ep.yaml -------------------------------------------------------------------------------- /configs/coco/swin/oneformer_swin_large_bs16_100ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/coco/swin/oneformer_swin_large_bs16_100ep.yaml -------------------------------------------------------------------------------- /configs/coco/swin/oneformer_swin_tiny_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/coco/swin/oneformer_swin_tiny_bs16_50ep.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/Base-Mapillary-UnifiedSegmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/Base-Mapillary-UnifiedSegmentation.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/convnext/cityscapes_pretrain_oneformer_convnext_large_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/convnext/cityscapes_pretrain_oneformer_convnext_large_bs16_300k.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/convnext/cityscapes_pretrain_oneformer_convnext_xlarge_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/convnext/cityscapes_pretrain_oneformer_convnext_xlarge_bs16_300k.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/convnext/oneformer_convnext_large_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/convnext/oneformer_convnext_large_bs16_300k.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/dinat/oneformer_dinat_large_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/dinat/oneformer_dinat_large_bs16_300k.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/oneformer_R50_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/oneformer_R50_bs16_300k.yaml -------------------------------------------------------------------------------- /configs/mapillary_vistas/swin/oneformer_swin_large_bs16_300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/configs/mapillary_vistas/swin/oneformer_swin_large_bs16_300k.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/ade20k_instance_catid_mapping.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/ade20k_instance_catid_mapping.txt -------------------------------------------------------------------------------- /datasets/ade20k_instance_imgCatIds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/ade20k_instance_imgCatIds.json -------------------------------------------------------------------------------- /datasets/custom_datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/custom_datasets/README.md -------------------------------------------------------------------------------- /datasets/custom_datasets/instance_coco_custom_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/custom_datasets/instance_coco_custom_dataset_mapper.py -------------------------------------------------------------------------------- /datasets/custom_datasets/instance_oneformer_custom_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/custom_datasets/instance_oneformer_custom_dataset_mapper.py -------------------------------------------------------------------------------- /datasets/custom_datasets/semantic_oneformer_custom_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/custom_datasets/semantic_oneformer_custom_dataset_mapper.py -------------------------------------------------------------------------------- /datasets/fg_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/fg_ids.py -------------------------------------------------------------------------------- /datasets/panoptic2detection_coco_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/panoptic2detection_coco_format.py -------------------------------------------------------------------------------- /datasets/panoptic_coco_categories.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/panoptic_coco_categories.json -------------------------------------------------------------------------------- /datasets/prepare_ade20k_ins_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/prepare_ade20k_ins_seg.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_pan_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/prepare_ade20k_pan_seg.py -------------------------------------------------------------------------------- /datasets/prepare_ade20k_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/prepare_ade20k_sem_seg.py -------------------------------------------------------------------------------- /datasets/prepare_coco_semantic_annos_from_panoptic_annos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/datasets/prepare_coco_semantic_annos_from_panoptic_annos.py -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/colormap.py -------------------------------------------------------------------------------- /demo/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/defaults.py -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/predictor.py -------------------------------------------------------------------------------- /demo/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/demo/visualizer.py -------------------------------------------------------------------------------- /images/oneformer.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/images/oneformer.svg -------------------------------------------------------------------------------- /images/plots.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/images/plots.svg -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/images/teaser.png -------------------------------------------------------------------------------- /images/teaser.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/images/teaser.svg -------------------------------------------------------------------------------- /oneformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/__init__.py -------------------------------------------------------------------------------- /oneformer/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/config.py -------------------------------------------------------------------------------- /oneformer/data/__init__.py: -------------------------------------------------------------------------------- 1 | from . import datasets 2 | -------------------------------------------------------------------------------- /oneformer/data/bpe_simple_vocab_16e6.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/bpe_simple_vocab_16e6.txt -------------------------------------------------------------------------------- /oneformer/data/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /oneformer/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/build.py -------------------------------------------------------------------------------- /oneformer/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /oneformer/data/dataset_mappers/coco_unified_new_baseline_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/dataset_mappers/coco_unified_new_baseline_dataset_mapper.py -------------------------------------------------------------------------------- /oneformer/data/dataset_mappers/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/dataset_mappers/dataset_mapper.py -------------------------------------------------------------------------------- /oneformer/data/dataset_mappers/oneformer_unified_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/dataset_mappers/oneformer_unified_dataset_mapper.py -------------------------------------------------------------------------------- /oneformer/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/__init__.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_ade20k_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_ade20k_instance.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_ade20k_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_ade20k_panoptic.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_cityscapes_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_cityscapes_panoptic.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_coco_panoptic2instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_coco_panoptic2instance.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_coco_panoptic_annos_semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_coco_panoptic_annos_semseg.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_mapillary_vistas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_mapillary_vistas.py -------------------------------------------------------------------------------- /oneformer/data/datasets/register_mapillary_vistas_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/datasets/register_mapillary_vistas_panoptic.py -------------------------------------------------------------------------------- /oneformer/data/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/data/tokenizer.py -------------------------------------------------------------------------------- /oneformer/datasetmapper_tta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/datasetmapper_tta.py -------------------------------------------------------------------------------- /oneformer/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/__init__.py -------------------------------------------------------------------------------- /oneformer/evaluation/cityscapes_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/cityscapes_evaluation.py -------------------------------------------------------------------------------- /oneformer/evaluation/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/coco_evaluator.py -------------------------------------------------------------------------------- /oneformer/evaluation/detection_coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/detection_coco_evaluator.py -------------------------------------------------------------------------------- /oneformer/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/evaluator.py -------------------------------------------------------------------------------- /oneformer/evaluation/instance_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/evaluation/instance_evaluation.py -------------------------------------------------------------------------------- /oneformer/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/__init__.py -------------------------------------------------------------------------------- /oneformer/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /oneformer/modeling/backbone/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/backbone/convnext.py -------------------------------------------------------------------------------- /oneformer/modeling/backbone/dinat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/backbone/dinat.py -------------------------------------------------------------------------------- /oneformer/modeling/backbone/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/backbone/swin.py -------------------------------------------------------------------------------- /oneformer/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/criterion.py -------------------------------------------------------------------------------- /oneformer/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/matcher.py -------------------------------------------------------------------------------- /oneformer/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /oneformer/modeling/meta_arch/oneformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/meta_arch/oneformer_head.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/fpn.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/msdeformattn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/msdeformattn.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/functions/__init__.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/make.sh -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/modules/__init__.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/setup.py -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/src/vision.cpp -------------------------------------------------------------------------------- /oneformer/modeling/pixel_decoder/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/pixel_decoder/ops/test.py -------------------------------------------------------------------------------- /oneformer/modeling/transformer_decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/transformer_decoder/__init__.py -------------------------------------------------------------------------------- /oneformer/modeling/transformer_decoder/oneformer_transformer_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/transformer_decoder/oneformer_transformer_decoder.py -------------------------------------------------------------------------------- /oneformer/modeling/transformer_decoder/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/transformer_decoder/position_encoding.py -------------------------------------------------------------------------------- /oneformer/modeling/transformer_decoder/text_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/transformer_decoder/text_transformer.py -------------------------------------------------------------------------------- /oneformer/modeling/transformer_decoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/modeling/transformer_decoder/transformer.py -------------------------------------------------------------------------------- /oneformer/oneformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/oneformer_model.py -------------------------------------------------------------------------------- /oneformer/test_time_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/test_time_augmentation.py -------------------------------------------------------------------------------- /oneformer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/utils/__init__.py -------------------------------------------------------------------------------- /oneformer/utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/utils/box_ops.py -------------------------------------------------------------------------------- /oneformer/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/utils/events.py -------------------------------------------------------------------------------- /oneformer/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/utils/misc.py -------------------------------------------------------------------------------- /oneformer/utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/oneformer/utils/pos_embed.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/README.md -------------------------------------------------------------------------------- /tools/analyze_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/analyze_model.py -------------------------------------------------------------------------------- /tools/calc_throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/calc_throughput.py -------------------------------------------------------------------------------- /tools/convert-pretrained-model-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/convert-pretrained-model-to-d2.py -------------------------------------------------------------------------------- /tools/convert-pretrained-nat-model-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/convert-pretrained-nat-model-to-d2.py -------------------------------------------------------------------------------- /tools/convert-torchvision-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/convert-torchvision-to-d2.py -------------------------------------------------------------------------------- /tools/setup_detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/setup_detectron2.py -------------------------------------------------------------------------------- /tools/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/trainers/trainer.py -------------------------------------------------------------------------------- /tools/trainers/trainer_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/tools/trainers/trainer_base.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/OneFormer/HEAD/train_net.py --------------------------------------------------------------------------------