├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── configs ├── config_MFNet.py ├── config_nyu.py ├── config_pst900.py └── config_sunrgbd.py ├── dataloader ├── RGBXDataset.py └── dataloader.py ├── engine ├── __init__.py ├── dist_test.py ├── engine.py ├── evaluator.py └── logger.py ├── eval.py ├── figs ├── overall_flops.png └── sigma.png ├── models ├── __init__.py ├── builder.py ├── decoders │ ├── MLPDecoder.py │ ├── MambaDecoder.py │ ├── UPernet.py │ ├── deeplabv3plus.py │ └── fcnhead.py ├── encoders │ ├── dual_segformer.py │ ├── dual_swin.py │ ├── dual_vmamba.py │ ├── selective_scan │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_core.cu │ │ │ │ ├── selective_scan_core_fwd2.cu │ │ │ │ ├── selective_scan_core_fwd3.cu │ │ │ │ ├── selective_scan_core_fwd4.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── readme.md │ │ ├── selective_scan │ │ │ ├── __init__.py │ │ │ └── selective_scan_interface.py │ │ ├── setup.py │ │ └── test_selective_scan.py │ └── vmamba.py └── net_utils.py ├── pretrained └── vmamba │ ├── vssmbase_dp06_ckpt_epoch_241.pth │ ├── vssmsmall_dp03_ckpt_epoch_238.pth │ └── vssmtiny_dp01_ckpt_epoch_292.pth ├── requirements.txt ├── train.py └── utils ├── __init__.py ├── calculate_flops_ConMB.py ├── calculate_params.py ├── init_func.py ├── load_utils.py ├── loss_opr.py ├── lr_policy.py ├── metric.py ├── pyt_utils.py ├── transforms.py └── visualize.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/README.md -------------------------------------------------------------------------------- /configs/config_MFNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/configs/config_MFNet.py -------------------------------------------------------------------------------- /configs/config_nyu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/configs/config_nyu.py -------------------------------------------------------------------------------- /configs/config_pst900.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/configs/config_pst900.py -------------------------------------------------------------------------------- /configs/config_sunrgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/configs/config_sunrgbd.py -------------------------------------------------------------------------------- /dataloader/RGBXDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/dataloader/RGBXDataset.py -------------------------------------------------------------------------------- /dataloader/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/dataloader/dataloader.py -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/dist_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/engine/dist_test.py -------------------------------------------------------------------------------- /engine/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/engine/engine.py -------------------------------------------------------------------------------- /engine/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/engine/evaluator.py -------------------------------------------------------------------------------- /engine/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/engine/logger.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/eval.py -------------------------------------------------------------------------------- /figs/overall_flops.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/figs/overall_flops.png -------------------------------------------------------------------------------- /figs/sigma.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/figs/sigma.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/builder.py -------------------------------------------------------------------------------- /models/decoders/MLPDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/decoders/MLPDecoder.py -------------------------------------------------------------------------------- /models/decoders/MambaDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/decoders/MambaDecoder.py -------------------------------------------------------------------------------- /models/decoders/UPernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/decoders/UPernet.py -------------------------------------------------------------------------------- /models/decoders/deeplabv3plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/decoders/deeplabv3plus.py -------------------------------------------------------------------------------- /models/decoders/fcnhead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/decoders/fcnhead.py -------------------------------------------------------------------------------- /models/encoders/dual_segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/dual_segformer.py -------------------------------------------------------------------------------- /models/encoders/dual_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/dual_swin.py -------------------------------------------------------------------------------- /models/encoders/dual_vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/dual_vmamba.py -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/reverse_scan.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/reverse_scan.cuh -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan.cpp -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan.h -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_bwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_bwd_kernel.cuh -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_common.h -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_core.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_core.cu -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd2.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd2.cu -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd3.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd3.cu -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd4.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_core_fwd4.cu -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/selective_scan_fwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/selective_scan_fwd_kernel.cuh -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/static_switch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/static_switch.h -------------------------------------------------------------------------------- /models/encoders/selective_scan/csrc/selective_scan/uninitialized_copy.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/csrc/selective_scan/uninitialized_copy.cuh -------------------------------------------------------------------------------- /models/encoders/selective_scan/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/readme.md -------------------------------------------------------------------------------- /models/encoders/selective_scan/selective_scan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/selective_scan/__init__.py -------------------------------------------------------------------------------- /models/encoders/selective_scan/selective_scan/selective_scan_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/selective_scan/selective_scan_interface.py -------------------------------------------------------------------------------- /models/encoders/selective_scan/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/setup.py -------------------------------------------------------------------------------- /models/encoders/selective_scan/test_selective_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/selective_scan/test_selective_scan.py -------------------------------------------------------------------------------- /models/encoders/vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/encoders/vmamba.py -------------------------------------------------------------------------------- /models/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/models/net_utils.py -------------------------------------------------------------------------------- /pretrained/vmamba/vssmbase_dp06_ckpt_epoch_241.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/pretrained/vmamba/vssmbase_dp06_ckpt_epoch_241.pth -------------------------------------------------------------------------------- /pretrained/vmamba/vssmsmall_dp03_ckpt_epoch_238.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/pretrained/vmamba/vssmsmall_dp03_ckpt_epoch_238.pth -------------------------------------------------------------------------------- /pretrained/vmamba/vssmtiny_dp01_ckpt_epoch_292.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/pretrained/vmamba/vssmtiny_dp01_ckpt_epoch_292.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/calculate_flops_ConMB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/calculate_flops_ConMB.py -------------------------------------------------------------------------------- /utils/calculate_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/calculate_params.py -------------------------------------------------------------------------------- /utils/init_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/init_func.py -------------------------------------------------------------------------------- /utils/load_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/load_utils.py -------------------------------------------------------------------------------- /utils/loss_opr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/loss_opr.py -------------------------------------------------------------------------------- /utils/lr_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/lr_policy.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/metric.py -------------------------------------------------------------------------------- /utils/pyt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/pyt_utils.py -------------------------------------------------------------------------------- /utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/transforms.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zifuwan/Sigma/HEAD/utils/visualize.py --------------------------------------------------------------------------------