├── .gitignore ├── LICENSE ├── README.md ├── cat_sam ├── datasets │ ├── base.py │ ├── kvasir.py │ ├── misc.py │ ├── sbu.py │ ├── transforms.py │ └── whu.py ├── models │ ├── decoders.py │ ├── encoders.py │ ├── modeling.py │ ├── module_lib.py │ └── segment_anything_ext │ │ ├── __init__.py │ │ ├── build_sam.py │ │ ├── modeling │ │ ├── __init__.py │ │ ├── common.py │ │ ├── image_encoder.py │ │ ├── mask_decoder.py │ │ ├── prompt_encoder.py │ │ ├── sam.py │ │ └── transformer.py │ │ └── utils │ │ ├── __init__.py │ │ └── transforms.py └── utils │ ├── evaluators.py │ └── samhq_misc.py ├── data ├── kvasir │ ├── README.md │ └── preprocess.py ├── sbu │ ├── README.md │ └── preprocess.py └── whu │ ├── README.md │ └── preprocess.py ├── figs ├── 1_shot.png └── CAT-SAM.png ├── misc.py ├── requirements.txt ├── setup.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | **/.DS_Store 2 | **/.idea/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/README.md -------------------------------------------------------------------------------- /cat_sam/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/base.py -------------------------------------------------------------------------------- /cat_sam/datasets/kvasir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/kvasir.py -------------------------------------------------------------------------------- /cat_sam/datasets/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/misc.py -------------------------------------------------------------------------------- /cat_sam/datasets/sbu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/sbu.py -------------------------------------------------------------------------------- /cat_sam/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/transforms.py -------------------------------------------------------------------------------- /cat_sam/datasets/whu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/datasets/whu.py -------------------------------------------------------------------------------- /cat_sam/models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/decoders.py -------------------------------------------------------------------------------- /cat_sam/models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/encoders.py -------------------------------------------------------------------------------- /cat_sam/models/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/modeling.py -------------------------------------------------------------------------------- /cat_sam/models/module_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/module_lib.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/__init__.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/build_sam.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/__init__.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/common.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/image_encoder.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/mask_decoder.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/sam.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/modeling/transformer.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/utils/__init__.py -------------------------------------------------------------------------------- /cat_sam/models/segment_anything_ext/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/models/segment_anything_ext/utils/transforms.py -------------------------------------------------------------------------------- /cat_sam/utils/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/utils/evaluators.py -------------------------------------------------------------------------------- /cat_sam/utils/samhq_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/cat_sam/utils/samhq_misc.py -------------------------------------------------------------------------------- /data/kvasir/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/kvasir/README.md -------------------------------------------------------------------------------- /data/kvasir/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/kvasir/preprocess.py -------------------------------------------------------------------------------- /data/sbu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/sbu/README.md -------------------------------------------------------------------------------- /data/sbu/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/sbu/preprocess.py -------------------------------------------------------------------------------- /data/whu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/whu/README.md -------------------------------------------------------------------------------- /data/whu/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/data/whu/preprocess.py -------------------------------------------------------------------------------- /figs/1_shot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/figs/1_shot.png -------------------------------------------------------------------------------- /figs/CAT-SAM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/figs/CAT-SAM.png -------------------------------------------------------------------------------- /misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/misc.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weihao1115/cat-sam/HEAD/train.py --------------------------------------------------------------------------------