├── README.md ├── create_dataset.py ├── images └── framework.png ├── losses.py ├── mobile_sam ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── automatic_mask_generator.cpython-38.pyc │ ├── automatic_mask_generator.cpython-39.pyc │ ├── build_sam.cpython-38.pyc │ ├── build_sam.cpython-39.pyc │ ├── predictor.cpython-38.pyc │ └── predictor.cpython-39.pyc ├── automatic_mask_generator.py ├── build_sam.py ├── modeling │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── common.cpython-38.pyc │ │ ├── common.cpython-39.pyc │ │ ├── image_encoder.cpython-38.pyc │ │ ├── image_encoder.cpython-39.pyc │ │ ├── mask_decoder.cpython-38.pyc │ │ ├── mask_decoder.cpython-39.pyc │ │ ├── prompt_encoder.cpython-38.pyc │ │ ├── prompt_encoder.cpython-39.pyc │ │ ├── sam.cpython-38.pyc │ │ ├── sam.cpython-39.pyc │ │ ├── tiny_vit_sam.cpython-38.pyc │ │ ├── tiny_vit_sam.cpython-39.pyc │ │ ├── transformer.cpython-38.pyc │ │ └── transformer.cpython-39.pyc │ ├── common.py │ ├── image_encoder.py │ ├── mask_decoder.py │ ├── prompt_encoder.py │ ├── sam.py │ ├── tiny_vit_sam.py │ └── transformer.py ├── predictor.py └── utils │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── amg.cpython-38.pyc │ ├── amg.cpython-39.pyc │ ├── transforms.cpython-38.pyc │ └── transforms.cpython-39.pyc │ ├── amg.py │ ├── onnx.py │ └── transforms.py ├── models ├── SEWeight.py ├── __pycache__ │ ├── SEWeight.cpython-38.pyc │ ├── SEWeight.cpython-39.pyc │ ├── attention.cpython-38.pyc │ ├── attention.cpython-39.pyc │ ├── common.cpython-38.pyc │ ├── common.cpython-39.pyc │ ├── groupmixattention.cpython-38.pyc │ └── groupmixattention.cpython-39.pyc ├── attention.py ├── common.py └── groupmixattention.py ├── network.py ├── optimizer.py ├── options.py ├── saver.py ├── test.py ├── train.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/README.md -------------------------------------------------------------------------------- /create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/create_dataset.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/images/framework.png -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/losses.py -------------------------------------------------------------------------------- /mobile_sam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__init__.py -------------------------------------------------------------------------------- /mobile_sam/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/automatic_mask_generator.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/automatic_mask_generator.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/automatic_mask_generator.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/automatic_mask_generator.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/build_sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/build_sam.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/build_sam.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/build_sam.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/predictor.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/predictor.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/__pycache__/predictor.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/__pycache__/predictor.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/automatic_mask_generator.py -------------------------------------------------------------------------------- /mobile_sam/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/build_sam.py -------------------------------------------------------------------------------- /mobile_sam/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__init__.py -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/common.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/common.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/image_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/image_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/image_encoder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/image_encoder.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/mask_decoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/mask_decoder.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/mask_decoder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/mask_decoder.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/prompt_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/prompt_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/prompt_encoder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/prompt_encoder.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/sam.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/sam.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/sam.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/tiny_vit_sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/tiny_vit_sam.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/tiny_vit_sam.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/tiny_vit_sam.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/transformer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/transformer.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/__pycache__/transformer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/__pycache__/transformer.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/common.py -------------------------------------------------------------------------------- /mobile_sam/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/image_encoder.py -------------------------------------------------------------------------------- /mobile_sam/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/mask_decoder.py -------------------------------------------------------------------------------- /mobile_sam/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /mobile_sam/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/sam.py -------------------------------------------------------------------------------- /mobile_sam/modeling/tiny_vit_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/tiny_vit_sam.py -------------------------------------------------------------------------------- /mobile_sam/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/modeling/transformer.py -------------------------------------------------------------------------------- /mobile_sam/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/predictor.py -------------------------------------------------------------------------------- /mobile_sam/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__init__.py -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/amg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/amg.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/amg.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/amg.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/transforms.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/transforms.cpython-38.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/__pycache__/transforms.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/__pycache__/transforms.cpython-39.pyc -------------------------------------------------------------------------------- /mobile_sam/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/amg.py -------------------------------------------------------------------------------- /mobile_sam/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/onnx.py -------------------------------------------------------------------------------- /mobile_sam/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/mobile_sam/utils/transforms.py -------------------------------------------------------------------------------- /models/SEWeight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/SEWeight.py -------------------------------------------------------------------------------- /models/__pycache__/SEWeight.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/SEWeight.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/SEWeight.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/SEWeight.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/attention.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/attention.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/attention.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/common.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/common.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/groupmixattention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/groupmixattention.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/groupmixattention.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/__pycache__/groupmixattention.cpython-39.pyc -------------------------------------------------------------------------------- /models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/attention.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/common.py -------------------------------------------------------------------------------- /models/groupmixattention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/models/groupmixattention.py -------------------------------------------------------------------------------- /network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/network.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/optimizer.py -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/options.py -------------------------------------------------------------------------------- /saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/saver.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxq-jnu/SpTFuse/HEAD/utils.py --------------------------------------------------------------------------------