├── LICENSE ├── README.md ├── SAM2UNeXT.py ├── dataset.py ├── eval.py ├── eval.sh ├── requirements.txt ├── sam2 ├── _C.pyd ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-38.pyc │ ├── build_sam.cpython-310.pyc │ ├── build_sam.cpython-38.pyc │ └── sam2_image_predictor.cpython-310.pyc ├── automatic_mask_generator.py ├── build_sam.py ├── csrc │ └── connected_components.cu ├── modeling │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── memory_attention.cpython-310.pyc │ │ ├── memory_attention.cpython-38.pyc │ │ ├── memory_encoder.cpython-310.pyc │ │ ├── memory_encoder.cpython-38.pyc │ │ ├── position_encoding.cpython-310.pyc │ │ ├── position_encoding.cpython-38.pyc │ │ ├── sam2_base.cpython-310.pyc │ │ ├── sam2_base.cpython-38.pyc │ │ ├── sam2_utils.cpython-310.pyc │ │ └── sam2_utils.cpython-38.pyc │ ├── backbones │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── hieradet.cpython-310.pyc │ │ │ ├── hieradet.cpython-38.pyc │ │ │ ├── image_encoder.cpython-310.pyc │ │ │ ├── image_encoder.cpython-38.pyc │ │ │ ├── utils.cpython-310.pyc │ │ │ └── utils.cpython-38.pyc │ │ ├── hieradet.py │ │ ├── image_encoder.py │ │ └── utils.py │ ├── memory_attention.py │ ├── memory_encoder.py │ ├── position_encoding.py │ ├── sam │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── mask_decoder.cpython-310.pyc │ │ │ ├── mask_decoder.cpython-38.pyc │ │ │ ├── prompt_encoder.cpython-310.pyc │ │ │ ├── prompt_encoder.cpython-38.pyc │ │ │ ├── transformer.cpython-310.pyc │ │ │ └── transformer.cpython-38.pyc │ │ ├── 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 │ ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-38.pyc │ ├── misc.cpython-310.pyc │ ├── misc.cpython-38.pyc │ └── transforms.cpython-310.pyc │ ├── amg.py │ ├── misc.py │ └── transforms.py ├── sam2_configs ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ └── __init__.cpython-38.pyc ├── sam2_hiera_b+.yaml ├── sam2_hiera_l.yaml ├── sam2_hiera_s.yaml └── sam2_hiera_t.yaml ├── sam2unext.png ├── test.py ├── test.sh ├── train.py └── train.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/README.md -------------------------------------------------------------------------------- /SAM2UNeXT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/SAM2UNeXT.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/dataset.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/eval.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/eval.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/requirements.txt -------------------------------------------------------------------------------- /sam2/_C.pyd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/_C.pyd -------------------------------------------------------------------------------- /sam2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__init__.py -------------------------------------------------------------------------------- /sam2/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/__pycache__/build_sam.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__pycache__/build_sam.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/__pycache__/build_sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__pycache__/build_sam.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/__pycache__/sam2_image_predictor.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/__pycache__/sam2_image_predictor.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/automatic_mask_generator.py -------------------------------------------------------------------------------- /sam2/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/build_sam.py -------------------------------------------------------------------------------- /sam2/csrc/connected_components.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/csrc/connected_components.cu -------------------------------------------------------------------------------- /sam2/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/memory_attention.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/memory_attention.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/memory_attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/memory_attention.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/memory_encoder.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/memory_encoder.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/memory_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/memory_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/position_encoding.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/position_encoding.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/position_encoding.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/position_encoding.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/sam2_base.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/sam2_base.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/sam2_base.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/sam2_base.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/sam2_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/sam2_utils.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/__pycache__/sam2_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/__pycache__/sam2_utils.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/hieradet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/hieradet.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/hieradet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/hieradet.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/image_encoder.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/image_encoder.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/image_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/image_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/utils.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/backbones/hieradet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/hieradet.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/image_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/backbones/utils.py -------------------------------------------------------------------------------- /sam2/modeling/memory_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/memory_attention.py -------------------------------------------------------------------------------- /sam2/modeling/memory_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/memory_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/position_encoding.py -------------------------------------------------------------------------------- /sam2/modeling/sam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__init__.py -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/mask_decoder.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/mask_decoder.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/mask_decoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/mask_decoder.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/prompt_encoder.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/prompt_encoder.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/prompt_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/prompt_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/transformer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/transformer.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/__pycache__/transformer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/__pycache__/transformer.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/modeling/sam/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/mask_decoder.py -------------------------------------------------------------------------------- /sam2/modeling/sam/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/prompt_encoder.py -------------------------------------------------------------------------------- /sam2/modeling/sam/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam/transformer.py -------------------------------------------------------------------------------- /sam2/modeling/sam2_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam2_base.py -------------------------------------------------------------------------------- /sam2/modeling/sam2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/modeling/sam2_utils.py -------------------------------------------------------------------------------- /sam2/sam2_image_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/sam2_image_predictor.py -------------------------------------------------------------------------------- /sam2/sam2_video_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/sam2_video_predictor.py -------------------------------------------------------------------------------- /sam2/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__init__.py -------------------------------------------------------------------------------- /sam2/utils/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/utils/__pycache__/misc.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__pycache__/misc.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/utils/__pycache__/misc.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__pycache__/misc.cpython-38.pyc -------------------------------------------------------------------------------- /sam2/utils/__pycache__/transforms.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/__pycache__/transforms.cpython-310.pyc -------------------------------------------------------------------------------- /sam2/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/amg.py -------------------------------------------------------------------------------- /sam2/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/misc.py -------------------------------------------------------------------------------- /sam2/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2/utils/transforms.py -------------------------------------------------------------------------------- /sam2_configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/__init__.py -------------------------------------------------------------------------------- /sam2_configs/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /sam2_configs/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_b+.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/sam2_hiera_b+.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/sam2_hiera_l.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/sam2_hiera_s.yaml -------------------------------------------------------------------------------- /sam2_configs/sam2_hiera_t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2_configs/sam2_hiera_t.yaml -------------------------------------------------------------------------------- /sam2unext.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/sam2unext.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/test.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/test.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WZH0120/SAM2-UNeXT/HEAD/train.sh --------------------------------------------------------------------------------