├── README.md ├── assets └── overview_.png ├── graphclustering ├── helpers │ ├── sam2object_utils.py │ ├── sam_utils.py │ └── visualize.py ├── linetimer.py ├── sam2object.py ├── sam2object_base.py └── scripts │ └── seg_scannet.sh └── segtrack ├── checkpoints └── download_ckpts.sh ├── dataprocess ├── extract_only_jpg.py └── get_posed_images.py ├── keyframe_extract_func.py ├── mask_convert.py ├── merge_video.py ├── sam2 ├── __init__.py ├── automatic_mask_generator.py ├── build_sam.py ├── configs │ ├── sam2.1 │ │ ├── sam2.1_hiera_b+.yaml │ │ ├── sam2.1_hiera_l.yaml │ │ ├── sam2.1_hiera_s.yaml │ │ └── sam2.1_hiera_t.yaml │ ├── sam2.1_training │ │ └── sam2.1_hiera_b+_MOSE_finetune.yaml │ └── sam2 │ │ ├── sam2_hiera_b+.yaml │ │ ├── sam2_hiera_l.yaml │ │ ├── sam2_hiera_s.yaml │ │ └── sam2_hiera_t.yaml ├── csrc │ └── connected_components.cu ├── 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_hiera_b+.yaml ├── sam2_hiera_l.yaml ├── sam2_hiera_s.yaml ├── sam2_hiera_t.yaml ├── sam2_image_predictor.py ├── sam2_video_predictor.py └── utils │ ├── __init__.py │ ├── amg.py │ ├── misc.py │ └── transforms.py ├── scannet_scene_val.txt ├── seg_tracking.py ├── setup.py └── utils ├── common_utils.py ├── mask_dictionary_model.py ├── supervision_utils.py ├── track_utils.py └── video_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/README.md -------------------------------------------------------------------------------- /assets/overview_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/assets/overview_.png -------------------------------------------------------------------------------- /graphclustering/helpers/sam2object_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/helpers/sam2object_utils.py -------------------------------------------------------------------------------- /graphclustering/helpers/sam_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/helpers/sam_utils.py -------------------------------------------------------------------------------- /graphclustering/helpers/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/helpers/visualize.py -------------------------------------------------------------------------------- /graphclustering/linetimer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/linetimer.py -------------------------------------------------------------------------------- /graphclustering/sam2object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/sam2object.py -------------------------------------------------------------------------------- /graphclustering/sam2object_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/sam2object_base.py -------------------------------------------------------------------------------- /graphclustering/scripts/seg_scannet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/graphclustering/scripts/seg_scannet.sh -------------------------------------------------------------------------------- /segtrack/checkpoints/download_ckpts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/checkpoints/download_ckpts.sh -------------------------------------------------------------------------------- /segtrack/dataprocess/extract_only_jpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/dataprocess/extract_only_jpg.py -------------------------------------------------------------------------------- /segtrack/dataprocess/get_posed_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/dataprocess/get_posed_images.py -------------------------------------------------------------------------------- /segtrack/keyframe_extract_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/keyframe_extract_func.py -------------------------------------------------------------------------------- /segtrack/mask_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/mask_convert.py -------------------------------------------------------------------------------- /segtrack/merge_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/merge_video.py -------------------------------------------------------------------------------- /segtrack/sam2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/__init__.py -------------------------------------------------------------------------------- /segtrack/sam2/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/automatic_mask_generator.py -------------------------------------------------------------------------------- /segtrack/sam2/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/build_sam.py -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2.1/sam2.1_hiera_b+.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2.1/sam2.1_hiera_b+.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2.1/sam2.1_hiera_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2.1/sam2.1_hiera_l.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2.1/sam2.1_hiera_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2.1/sam2.1_hiera_s.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2.1/sam2.1_hiera_t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2.1/sam2.1_hiera_t.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2.1_training/sam2.1_hiera_b+_MOSE_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2.1_training/sam2.1_hiera_b+_MOSE_finetune.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2/sam2_hiera_b+.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2/sam2_hiera_b+.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2/sam2_hiera_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2/sam2_hiera_l.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2/sam2_hiera_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2/sam2_hiera_s.yaml -------------------------------------------------------------------------------- /segtrack/sam2/configs/sam2/sam2_hiera_t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/configs/sam2/sam2_hiera_t.yaml -------------------------------------------------------------------------------- /segtrack/sam2/csrc/connected_components.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/csrc/connected_components.cu -------------------------------------------------------------------------------- /segtrack/sam2/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/__init__.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/backbones/hieradet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/backbones/hieradet.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/backbones/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/backbones/image_encoder.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/backbones/utils.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/memory_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/memory_attention.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/memory_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/memory_encoder.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/position_encoding.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam/__init__.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam/mask_decoder.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam/prompt_encoder.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam/transformer.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam2_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam2_base.py -------------------------------------------------------------------------------- /segtrack/sam2/modeling/sam2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/modeling/sam2_utils.py -------------------------------------------------------------------------------- /segtrack/sam2/sam2_hiera_b+.yaml: -------------------------------------------------------------------------------- 1 | configs/sam2/sam2_hiera_b+.yaml -------------------------------------------------------------------------------- /segtrack/sam2/sam2_hiera_l.yaml: -------------------------------------------------------------------------------- 1 | configs/sam2/sam2_hiera_l.yaml -------------------------------------------------------------------------------- /segtrack/sam2/sam2_hiera_s.yaml: -------------------------------------------------------------------------------- 1 | configs/sam2/sam2_hiera_s.yaml -------------------------------------------------------------------------------- /segtrack/sam2/sam2_hiera_t.yaml: -------------------------------------------------------------------------------- 1 | configs/sam2/sam2_hiera_t.yaml -------------------------------------------------------------------------------- /segtrack/sam2/sam2_image_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/sam2_image_predictor.py -------------------------------------------------------------------------------- /segtrack/sam2/sam2_video_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/sam2_video_predictor.py -------------------------------------------------------------------------------- /segtrack/sam2/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/utils/__init__.py -------------------------------------------------------------------------------- /segtrack/sam2/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/utils/amg.py -------------------------------------------------------------------------------- /segtrack/sam2/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/utils/misc.py -------------------------------------------------------------------------------- /segtrack/sam2/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/sam2/utils/transforms.py -------------------------------------------------------------------------------- /segtrack/scannet_scene_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/scannet_scene_val.txt -------------------------------------------------------------------------------- /segtrack/seg_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/seg_tracking.py -------------------------------------------------------------------------------- /segtrack/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/setup.py -------------------------------------------------------------------------------- /segtrack/utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/utils/common_utils.py -------------------------------------------------------------------------------- /segtrack/utils/mask_dictionary_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/utils/mask_dictionary_model.py -------------------------------------------------------------------------------- /segtrack/utils/supervision_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/utils/supervision_utils.py -------------------------------------------------------------------------------- /segtrack/utils/track_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/utils/track_utils.py -------------------------------------------------------------------------------- /segtrack/utils/video_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jihuaizhaohd/SAM2Object/HEAD/segtrack/utils/video_utils.py --------------------------------------------------------------------------------