├── .gitignore ├── LICENSE ├── README.md ├── ckpts └── README.md ├── configs ├── clip │ ├── simseg.vit-b.yaml │ └── simseg.vit-s.yaml └── linear_prob │ └── imagenet.yaml ├── data ├── README.md └── label_category │ ├── coco_stuff.txt │ ├── pascal_context.txt │ └── pascal_voc.txt ├── docs └── cvpr23_simseg.png ├── launch.py ├── requirements.txt ├── simseg ├── __init__.py ├── core │ ├── __init__.py │ ├── config.py │ ├── hooks │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── dist.py │ │ ├── freeze_bn.py │ │ ├── hook.py │ │ ├── log.py │ │ ├── optimizer.py │ │ ├── utils.py │ │ └── wandb_hook.py │ ├── initial.py │ ├── optimizer │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── lars.py │ │ └── lr_scheduler.py │ └── runners │ │ ├── __init__.py │ │ ├── base_runner.py │ │ ├── builder.py │ │ └── epoch_runner.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── clip │ │ ├── __init__.py │ │ ├── clip_dataset.py │ │ └── utils.py │ ├── linear_prob │ │ ├── __init__.py │ │ └── linear_dataset.py │ └── seg │ │ ├── __init__.py │ │ ├── seg_dataset.py │ │ └── utils.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── builder.py │ │ └── mml │ │ │ ├── __init__.py │ │ │ ├── huggingface_builder.py │ │ │ ├── timm_builder.py │ │ │ └── vit_builder.py │ ├── components │ │ ├── __init__.py │ │ ├── grad_rev.py │ │ ├── normalization.py │ │ ├── pooling.py │ │ └── projection.py │ ├── criteria │ │ ├── __init__.py │ │ ├── builder.py │ │ └── losses │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── mml_loss.py │ │ │ └── utils.py │ └── pipelines │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── clip.py │ │ └── linear_prob.py ├── tasks │ ├── __init__.py │ ├── clip │ │ ├── __init__.py │ │ ├── clip_bsgs_runner.py │ │ ├── clip_runner.py │ │ ├── config.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── checkpoint.py │ │ │ ├── eval.py │ │ │ ├── optimizer.py │ │ │ └── utils.py │ │ └── train.py │ └── linear_prob │ │ ├── __init__.py │ │ ├── config.py │ │ ├── hooks │ │ ├── __init__.py │ │ ├── eval.py │ │ └── utils.py │ │ ├── linear_runner.py │ │ └── train.py ├── transforms │ ├── __init__.py │ └── mml │ │ ├── __init__.py │ │ ├── auto_augment.py │ │ ├── color_distortion.py │ │ ├── gaussian_blur.py │ │ ├── random_erasing.py │ │ └── transforms.py ├── utils │ ├── __init__.py │ ├── checkpoint_utils.py │ ├── collections.py │ ├── context.py │ ├── dist.py │ ├── interpolate_pe.py │ ├── logger.py │ ├── metrics.py │ ├── misc.py │ ├── prompt.py │ └── registry.py └── version.py └── tools ├── convert_datasets ├── coco_stuff164k.py └── pascal_context.py ├── retrieval_evaluation.py └── seg_evaluation.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/README.md -------------------------------------------------------------------------------- /ckpts/README.md: -------------------------------------------------------------------------------- 1 | Please save pre-trained models here for evaluation. 2 | -------------------------------------------------------------------------------- /configs/clip/simseg.vit-b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/configs/clip/simseg.vit-b.yaml -------------------------------------------------------------------------------- /configs/clip/simseg.vit-s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/configs/clip/simseg.vit-s.yaml -------------------------------------------------------------------------------- /configs/linear_prob/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/configs/linear_prob/imagenet.yaml -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- 1 | Place your training and validation datasets here. -------------------------------------------------------------------------------- /data/label_category/coco_stuff.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/data/label_category/coco_stuff.txt -------------------------------------------------------------------------------- /data/label_category/pascal_context.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/data/label_category/pascal_context.txt -------------------------------------------------------------------------------- /data/label_category/pascal_voc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/data/label_category/pascal_voc.txt -------------------------------------------------------------------------------- /docs/cvpr23_simseg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/docs/cvpr23_simseg.png -------------------------------------------------------------------------------- /launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/launch.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/requirements.txt -------------------------------------------------------------------------------- /simseg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/__init__.py -------------------------------------------------------------------------------- /simseg/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/__init__.py -------------------------------------------------------------------------------- /simseg/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/config.py -------------------------------------------------------------------------------- /simseg/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/__init__.py -------------------------------------------------------------------------------- /simseg/core/hooks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/checkpoint.py -------------------------------------------------------------------------------- /simseg/core/hooks/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/dist.py -------------------------------------------------------------------------------- /simseg/core/hooks/freeze_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/freeze_bn.py -------------------------------------------------------------------------------- /simseg/core/hooks/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/hook.py -------------------------------------------------------------------------------- /simseg/core/hooks/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/log.py -------------------------------------------------------------------------------- /simseg/core/hooks/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/optimizer.py -------------------------------------------------------------------------------- /simseg/core/hooks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/utils.py -------------------------------------------------------------------------------- /simseg/core/hooks/wandb_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/hooks/wandb_hook.py -------------------------------------------------------------------------------- /simseg/core/initial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/initial.py -------------------------------------------------------------------------------- /simseg/core/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/optimizer/__init__.py -------------------------------------------------------------------------------- /simseg/core/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/optimizer/builder.py -------------------------------------------------------------------------------- /simseg/core/optimizer/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/optimizer/lars.py -------------------------------------------------------------------------------- /simseg/core/optimizer/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/optimizer/lr_scheduler.py -------------------------------------------------------------------------------- /simseg/core/runners/__init__.py: -------------------------------------------------------------------------------- 1 | from .builder import RUNNER -------------------------------------------------------------------------------- /simseg/core/runners/base_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/runners/base_runner.py -------------------------------------------------------------------------------- /simseg/core/runners/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/runners/builder.py -------------------------------------------------------------------------------- /simseg/core/runners/epoch_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/core/runners/epoch_runner.py -------------------------------------------------------------------------------- /simseg/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/__init__.py -------------------------------------------------------------------------------- /simseg/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/builder.py -------------------------------------------------------------------------------- /simseg/datasets/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip_dataset import * -------------------------------------------------------------------------------- /simseg/datasets/clip/clip_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/clip/clip_dataset.py -------------------------------------------------------------------------------- /simseg/datasets/clip/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/clip/utils.py -------------------------------------------------------------------------------- /simseg/datasets/linear_prob/__init__.py: -------------------------------------------------------------------------------- 1 | from .linear_dataset import * -------------------------------------------------------------------------------- /simseg/datasets/linear_prob/linear_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/linear_prob/linear_dataset.py -------------------------------------------------------------------------------- /simseg/datasets/seg/__init__.py: -------------------------------------------------------------------------------- 1 | from .seg_dataset import * -------------------------------------------------------------------------------- /simseg/datasets/seg/seg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/datasets/seg/seg_dataset.py -------------------------------------------------------------------------------- /simseg/datasets/seg/utils.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simseg/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/__init__.py -------------------------------------------------------------------------------- /simseg/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/__init__.py -------------------------------------------------------------------------------- /simseg/models/backbones/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/builder.py -------------------------------------------------------------------------------- /simseg/models/backbones/mml/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/mml/__init__.py -------------------------------------------------------------------------------- /simseg/models/backbones/mml/huggingface_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/mml/huggingface_builder.py -------------------------------------------------------------------------------- /simseg/models/backbones/mml/timm_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/mml/timm_builder.py -------------------------------------------------------------------------------- /simseg/models/backbones/mml/vit_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/backbones/mml/vit_builder.py -------------------------------------------------------------------------------- /simseg/models/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/components/__init__.py -------------------------------------------------------------------------------- /simseg/models/components/grad_rev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/components/grad_rev.py -------------------------------------------------------------------------------- /simseg/models/components/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/components/normalization.py -------------------------------------------------------------------------------- /simseg/models/components/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/components/pooling.py -------------------------------------------------------------------------------- /simseg/models/components/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/components/projection.py -------------------------------------------------------------------------------- /simseg/models/criteria/__init__.py: -------------------------------------------------------------------------------- 1 | from .builder import CRITERION -------------------------------------------------------------------------------- /simseg/models/criteria/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/criteria/builder.py -------------------------------------------------------------------------------- /simseg/models/criteria/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/criteria/losses/__init__.py -------------------------------------------------------------------------------- /simseg/models/criteria/losses/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/criteria/losses/builder.py -------------------------------------------------------------------------------- /simseg/models/criteria/losses/mml_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/criteria/losses/mml_loss.py -------------------------------------------------------------------------------- /simseg/models/criteria/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/criteria/losses/utils.py -------------------------------------------------------------------------------- /simseg/models/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/pipelines/__init__.py -------------------------------------------------------------------------------- /simseg/models/pipelines/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/pipelines/builder.py -------------------------------------------------------------------------------- /simseg/models/pipelines/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/pipelines/clip.py -------------------------------------------------------------------------------- /simseg/models/pipelines/linear_prob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/models/pipelines/linear_prob.py -------------------------------------------------------------------------------- /simseg/tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simseg/tasks/clip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simseg/tasks/clip/clip_bsgs_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/clip_bsgs_runner.py -------------------------------------------------------------------------------- /simseg/tasks/clip/clip_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/clip_runner.py -------------------------------------------------------------------------------- /simseg/tasks/clip/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/config.py -------------------------------------------------------------------------------- /simseg/tasks/clip/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/hooks/__init__.py -------------------------------------------------------------------------------- /simseg/tasks/clip/hooks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/hooks/checkpoint.py -------------------------------------------------------------------------------- /simseg/tasks/clip/hooks/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/hooks/eval.py -------------------------------------------------------------------------------- /simseg/tasks/clip/hooks/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/hooks/optimizer.py -------------------------------------------------------------------------------- /simseg/tasks/clip/hooks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/hooks/utils.py -------------------------------------------------------------------------------- /simseg/tasks/clip/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/clip/train.py -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/linear_prob/config.py -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/hooks/__init__.py: -------------------------------------------------------------------------------- 1 | from .eval import LinearEvalHook -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/hooks/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/linear_prob/hooks/eval.py -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/hooks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/linear_prob/hooks/utils.py -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/linear_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/linear_prob/linear_runner.py -------------------------------------------------------------------------------- /simseg/tasks/linear_prob/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/tasks/linear_prob/train.py -------------------------------------------------------------------------------- /simseg/transforms/__init__.py: -------------------------------------------------------------------------------- 1 | from .mml import * 2 | -------------------------------------------------------------------------------- /simseg/transforms/mml/__init__.py: -------------------------------------------------------------------------------- 1 | from .transforms import * -------------------------------------------------------------------------------- /simseg/transforms/mml/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/transforms/mml/auto_augment.py -------------------------------------------------------------------------------- /simseg/transforms/mml/color_distortion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/transforms/mml/color_distortion.py -------------------------------------------------------------------------------- /simseg/transforms/mml/gaussian_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/transforms/mml/gaussian_blur.py -------------------------------------------------------------------------------- /simseg/transforms/mml/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/transforms/mml/random_erasing.py -------------------------------------------------------------------------------- /simseg/transforms/mml/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/transforms/mml/transforms.py -------------------------------------------------------------------------------- /simseg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/__init__.py -------------------------------------------------------------------------------- /simseg/utils/checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/checkpoint_utils.py -------------------------------------------------------------------------------- /simseg/utils/collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/collections.py -------------------------------------------------------------------------------- /simseg/utils/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/context.py -------------------------------------------------------------------------------- /simseg/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/dist.py -------------------------------------------------------------------------------- /simseg/utils/interpolate_pe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/interpolate_pe.py -------------------------------------------------------------------------------- /simseg/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/logger.py -------------------------------------------------------------------------------- /simseg/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/metrics.py -------------------------------------------------------------------------------- /simseg/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/misc.py -------------------------------------------------------------------------------- /simseg/utils/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/prompt.py -------------------------------------------------------------------------------- /simseg/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/simseg/utils/registry.py -------------------------------------------------------------------------------- /simseg/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.1.0' 2 | -------------------------------------------------------------------------------- /tools/convert_datasets/coco_stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/tools/convert_datasets/coco_stuff164k.py -------------------------------------------------------------------------------- /tools/convert_datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/tools/convert_datasets/pascal_context.py -------------------------------------------------------------------------------- /tools/retrieval_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/tools/retrieval_evaluation.py -------------------------------------------------------------------------------- /tools/seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/muyangyi/SimSeg/HEAD/tools/seg_evaluation.py --------------------------------------------------------------------------------