├── LICENSE ├── README.md ├── dataset_names ├── __init__.py ├── ade20k_150_with_prompt_eng.txt ├── ade20k_847_with_prompt_eng.txt ├── cityscapes_with_prompt_eng.txt ├── coco_panoptic_with_prompt_eng.txt ├── imagenet_folder.txt ├── lvis_1203_with_prompt_eng.txt ├── openseg_classes.py ├── pascal_ctx_459_with_prompt_eng.txt └── v3det_cat_names.txt ├── dataset_preparation ├── README.md ├── convert_ade_847_to_wds.py ├── convert_cityscapes_panoptic_to_wds.py ├── convert_coco_panoptic_to_wds.py ├── convert_lvis_to_wds.py ├── convert_partimagenet_to_wds.py ├── convert_pascalpart_to_wds.py ├── convert_pc459_to_wds copy.py └── convert_v3det_to_wds.py ├── demo_with_sam.py ├── imgs ├── demo.jpg ├── task.png ├── teaser.png ├── vis.png ├── vis2.png └── vis3.png ├── interactive_demo.ipynb ├── modeling ├── __init__.py ├── factory.py └── osm.py ├── requirements.txt ├── test ├── evaluate_pred.py └── generate_pred.py ├── train ├── __init__.py ├── data.py ├── data_utils.py ├── distributed.py ├── train.py └── train_utils.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/README.md -------------------------------------------------------------------------------- /dataset_names/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/__init__.py -------------------------------------------------------------------------------- /dataset_names/ade20k_150_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/ade20k_150_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/ade20k_847_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/ade20k_847_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/cityscapes_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/cityscapes_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/coco_panoptic_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/coco_panoptic_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/imagenet_folder.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/imagenet_folder.txt -------------------------------------------------------------------------------- /dataset_names/lvis_1203_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/lvis_1203_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/openseg_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/openseg_classes.py -------------------------------------------------------------------------------- /dataset_names/pascal_ctx_459_with_prompt_eng.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/pascal_ctx_459_with_prompt_eng.txt -------------------------------------------------------------------------------- /dataset_names/v3det_cat_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_names/v3det_cat_names.txt -------------------------------------------------------------------------------- /dataset_preparation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/README.md -------------------------------------------------------------------------------- /dataset_preparation/convert_ade_847_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_ade_847_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_cityscapes_panoptic_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_cityscapes_panoptic_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_coco_panoptic_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_coco_panoptic_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_lvis_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_lvis_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_partimagenet_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_partimagenet_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_pascalpart_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_pascalpart_to_wds.py -------------------------------------------------------------------------------- /dataset_preparation/convert_pc459_to_wds copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_pc459_to_wds copy.py -------------------------------------------------------------------------------- /dataset_preparation/convert_v3det_to_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/dataset_preparation/convert_v3det_to_wds.py -------------------------------------------------------------------------------- /demo_with_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/demo_with_sam.py -------------------------------------------------------------------------------- /imgs/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/demo.jpg -------------------------------------------------------------------------------- /imgs/task.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/task.png -------------------------------------------------------------------------------- /imgs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/teaser.png -------------------------------------------------------------------------------- /imgs/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/vis.png -------------------------------------------------------------------------------- /imgs/vis2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/vis2.png -------------------------------------------------------------------------------- /imgs/vis3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/imgs/vis3.png -------------------------------------------------------------------------------- /interactive_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/interactive_demo.ipynb -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/modeling/__init__.py -------------------------------------------------------------------------------- /modeling/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/modeling/factory.py -------------------------------------------------------------------------------- /modeling/osm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/modeling/osm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/evaluate_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/test/evaluate_pred.py -------------------------------------------------------------------------------- /test/generate_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/test/generate_pred.py -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/__init__.py -------------------------------------------------------------------------------- /train/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/data.py -------------------------------------------------------------------------------- /train/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/data_utils.py -------------------------------------------------------------------------------- /train/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/distributed.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/train.py -------------------------------------------------------------------------------- /train/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/train/train_utils.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bytedance/OmniScient-Model/HEAD/utils.py --------------------------------------------------------------------------------