├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── docs └── images │ └── example.jpg ├── install.py ├── local_groundingdino ├── datasets │ ├── __init__.py │ └── transforms.py ├── models │ ├── GroundingDINO │ │ ├── __init__.py │ │ ├── backbone │ │ │ ├── __init__.py │ │ │ ├── backbone.py │ │ │ ├── position_encoding.py │ │ │ └── swin_transformer.py │ │ ├── bertwarper.py │ │ ├── fuse_modules.py │ │ ├── groundingdino.py │ │ ├── ms_deform_attn.py │ │ ├── transformer.py │ │ ├── transformer_vanilla.py │ │ └── utils.py │ ├── __init__.py │ └── registry.py └── util │ ├── __init__.py │ ├── box_ops.py │ ├── get_tokenlizer.py │ ├── inference.py │ ├── misc.py │ ├── slconfig.py │ ├── slio.py │ └── utils.py ├── node.py ├── requirements.txt └── sam_hq ├── automatic.py ├── build_sam_hq.py ├── modeling ├── image_encoder.py ├── mask_decoder_hq.py └── tiny_vit.py └── predictor.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/__init__.py -------------------------------------------------------------------------------- /docs/images/example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/docs/images/example.jpg -------------------------------------------------------------------------------- /install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/install.py -------------------------------------------------------------------------------- /local_groundingdino/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /local_groundingdino/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/datasets/transforms.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/__init__.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/backbone/__init__.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/backbone/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/backbone/backbone.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/backbone/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/backbone/position_encoding.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/bertwarper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/bertwarper.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/fuse_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/fuse_modules.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/groundingdino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/groundingdino.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/ms_deform_attn.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/transformer.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/transformer_vanilla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/transformer_vanilla.py -------------------------------------------------------------------------------- /local_groundingdino/models/GroundingDINO/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/GroundingDINO/utils.py -------------------------------------------------------------------------------- /local_groundingdino/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/__init__.py -------------------------------------------------------------------------------- /local_groundingdino/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/models/registry.py -------------------------------------------------------------------------------- /local_groundingdino/util/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /local_groundingdino/util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/box_ops.py -------------------------------------------------------------------------------- /local_groundingdino/util/get_tokenlizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/get_tokenlizer.py -------------------------------------------------------------------------------- /local_groundingdino/util/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/inference.py -------------------------------------------------------------------------------- /local_groundingdino/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/misc.py -------------------------------------------------------------------------------- /local_groundingdino/util/slconfig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/slconfig.py -------------------------------------------------------------------------------- /local_groundingdino/util/slio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/slio.py -------------------------------------------------------------------------------- /local_groundingdino/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/local_groundingdino/util/utils.py -------------------------------------------------------------------------------- /node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/node.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | segment_anything 2 | timm 3 | addict 4 | yapf -------------------------------------------------------------------------------- /sam_hq/automatic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/automatic.py -------------------------------------------------------------------------------- /sam_hq/build_sam_hq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/build_sam_hq.py -------------------------------------------------------------------------------- /sam_hq/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/modeling/image_encoder.py -------------------------------------------------------------------------------- /sam_hq/modeling/mask_decoder_hq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/modeling/mask_decoder_hq.py -------------------------------------------------------------------------------- /sam_hq/modeling/tiny_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/modeling/tiny_vit.py -------------------------------------------------------------------------------- /sam_hq/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/storyicon/comfyui_segment_anything/HEAD/sam_hq/predictor.py --------------------------------------------------------------------------------