├── .gitattributes ├── .github └── workflows │ └── publish.yml ├── .gitignore ├── LICENSE ├── __init__.py ├── example_workflows ├── florence_segment_2.json ├── image_batch_bbox_segment.json └── points_segment_video_example.json ├── load_model.py ├── nodes.py ├── pyproject.toml ├── readme.md ├── sam2 ├── __init__.py ├── automatic_mask_generator.py ├── modeling │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── hieradet.py │ │ ├── image_encoder.py │ │ └── utils.py │ ├── memory_attention.py │ ├── memory_encoder.py │ ├── position_encoding.py │ ├── sam │ │ ├── __init__.py │ │ ├── mask_decoder.py │ │ ├── prompt_encoder.py │ │ └── transformer.py │ ├── sam2_base.py │ └── sam2_utils.py ├── sam2_image_predictor.py ├── sam2_video_predictor.py └── utils │ ├── __init__.py │ ├── amg.py │ ├── misc.py │ └── transforms.py └── sam2_configs ├── __init__.py ├── sam2.1_hiera_b+.yaml ├── sam2.1_hiera_l.yaml ├── sam2.1_hiera_s.yaml ├── sam2.1_hiera_t.yaml ├── sam2_hiera_b+.yaml ├── sam2_hiera_l.yaml ├── sam2_hiera_s.yaml └── sam2_hiera_t.yaml /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/LICENSE -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/__init__.py -------------------------------------------------------------------------------- /example_workflows/florence_segment_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/example_workflows/florence_segment_2.json -------------------------------------------------------------------------------- /example_workflows/image_batch_bbox_segment.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/example_workflows/image_batch_bbox_segment.json -------------------------------------------------------------------------------- /example_workflows/points_segment_video_example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/example_workflows/points_segment_video_example.json -------------------------------------------------------------------------------- /load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/load_model.py -------------------------------------------------------------------------------- /nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/nodes.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/readme.md -------------------------------------------------------------------------------- /sam2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/__init__.py -------------------------------------------------------------------------------- /sam2/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/automatic_mask_generator.py -------------------------------------------------------------------------------- /sam2/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/hieradet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/backbones/hieradet.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/backbones/image_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/backbones/utils.py -------------------------------------------------------------------------------- /sam2/modeling/memory_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/memory_attention.py -------------------------------------------------------------------------------- /sam2/modeling/memory_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/memory_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/position_encoding.py -------------------------------------------------------------------------------- /sam2/modeling/sam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/sam/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam/mask_decoder.py -------------------------------------------------------------------------------- /sam2/modeling/sam/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam/prompt_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/sam/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam/transformer.py -------------------------------------------------------------------------------- /sam2/modeling/sam2_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam2_base.py -------------------------------------------------------------------------------- /sam2/modeling/sam2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/modeling/sam2_utils.py -------------------------------------------------------------------------------- /sam2/sam2_image_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/sam2_image_predictor.py -------------------------------------------------------------------------------- /sam2/sam2_video_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/sam2_video_predictor.py -------------------------------------------------------------------------------- /sam2/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/utils/__init__.py -------------------------------------------------------------------------------- /sam2/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/utils/amg.py -------------------------------------------------------------------------------- /sam2/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/utils/misc.py -------------------------------------------------------------------------------- /sam2/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2/utils/transforms.py -------------------------------------------------------------------------------- /sam2_configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/__init__.py -------------------------------------------------------------------------------- /sam2_configs/sam2.1_hiera_b+.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2.1_hiera_b+.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2.1_hiera_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2.1_hiera_l.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2.1_hiera_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2.1_hiera_s.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2.1_hiera_t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2.1_hiera_t.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_b+.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2_hiera_b+.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2_hiera_l.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2_hiera_s.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kijai/ComfyUI-segment-anything-2/HEAD/sam2_configs/sam2_hiera_t.yaml --------------------------------------------------------------------------------