├── LICENSE ├── detection ├── config │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k_instance.py │ │ │ ├── ade20k_panoptic.py │ │ │ ├── ade20k_semantic.py │ │ │ ├── cityscapes_detection.py │ │ │ ├── cityscapes_instance.py │ │ │ ├── coco_caption.py │ │ │ ├── coco_detection.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_instance_semantic.py │ │ │ ├── coco_panoptic.py │ │ │ ├── coco_semantic.py │ │ │ ├── deepfashion.py │ │ │ ├── dsdl.py │ │ │ ├── isaid_instance.py │ │ │ ├── lvis_v0.5_instance.py │ │ │ ├── lvis_v1_instance.py │ │ │ ├── mot_challenge.py │ │ │ ├── mot_challenge_det.py │ │ │ ├── mot_challenge_reid.py │ │ │ ├── objects365v1_detection.py │ │ │ ├── objects365v2_detection.py │ │ │ ├── openimages_detection.py │ │ │ ├── refcoco+.py │ │ │ ├── refcoco.py │ │ │ ├── refcocog.py │ │ │ ├── semi_coco_detection.py │ │ │ ├── v3det.py │ │ │ ├── voc0712.py │ │ │ ├── wider_face.py │ │ │ └── youtube_vis.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── cascade-mask-rcnn_r50_fpn.py │ │ │ ├── cascade-rcnn_r50_fpn.py │ │ │ ├── fast-rcnn_r50_fpn.py │ │ │ ├── faster-rcnn_r50-caffe-c4.py │ │ │ ├── faster-rcnn_r50-caffe-dc5.py │ │ │ ├── faster-rcnn_r50_fpn.py │ │ │ ├── mask-rcnn_r50-caffe-c4.py │ │ │ ├── mask-rcnn_r50_fpn.py │ │ │ ├── retinanet_r50_fpn.py │ │ │ ├── rpn_r50-caffe-c4.py │ │ │ ├── rpn_r50_fpn.py │ │ │ └── ssd300.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ ├── mask-rcnn_swin-t-p4-w7_fpn_1x_coco.py │ ├── maskrcnn_sparx_mamba_b_fpn_1x_coco.py │ ├── maskrcnn_sparx_mamba_b_fpn_3x_coco.py │ ├── maskrcnn_sparx_mamba_s_fpn_1x_coco.py │ ├── maskrcnn_sparx_mamba_s_fpn_3x_coco.py │ ├── maskrcnn_sparx_mamba_t_fpn_1x_coco.py │ └── maskrcnn_sparx_mamba_t_fpn_3x_coco.py ├── models │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-311.pyc │ │ ├── csm_triton.cpython-311.pyc │ │ └── sparx_mamba.cpython-311.pyc │ ├── csm_triton.py │ └── sparx_mamba.py ├── readme.md ├── scripts │ ├── dist_test.sh │ └── dist_train.sh ├── test.py └── train.py ├── images └── arch.jpg ├── kernels └── selective_scan │ ├── README.md │ ├── csrc │ └── selective_scan │ │ ├── cub_extra.cuh │ │ ├── cus │ │ ├── selective_scan.cpp │ │ ├── selective_scan_bwd_kernel.cuh │ │ ├── selective_scan_core_bwd.cu │ │ ├── selective_scan_core_fwd.cu │ │ └── selective_scan_fwd_kernel.cuh │ │ ├── cusoflex │ │ ├── selective_scan_bwd_kernel_oflex.cuh │ │ ├── selective_scan_core_bwd.cu │ │ ├── selective_scan_core_fwd.cu │ │ ├── selective_scan_fwd_kernel_oflex.cuh │ │ ├── selective_scan_oflex.cpp │ │ └── selective_scan_oflex.h │ │ ├── reverse_scan.cuh │ │ ├── selective_scan.h │ │ ├── selective_scan_common.h │ │ ├── static_switch.h │ │ └── uninitialized_copy.cuh │ ├── setup.py │ ├── test_selective_scan.py │ ├── test_selective_scan_easy.py │ └── test_selective_scan_speed.py ├── models ├── __init__.py ├── csm_triton.py └── sparx_mamba.py ├── readme.md ├── scripts ├── train_sparx_mamba_b.sh ├── train_sparx_mamba_s.sh └── train_sparx_mamba_t.sh ├── segmentation ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ └── ade20k.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── ann_r50-d8.py │ │ │ ├── apcnet_r50-d8.py │ │ │ ├── bisenetv1_r18-d32.py │ │ │ ├── bisenetv2.py │ │ │ ├── ccnet_r50-d8.py │ │ │ ├── cgnet.py │ │ │ ├── danet_r50-d8.py │ │ │ ├── deeplabv3_r50-d8.py │ │ │ ├── deeplabv3_unet_s5-d16.py │ │ │ ├── deeplabv3plus_r50-d8.py │ │ │ ├── dmnet_r50-d8.py │ │ │ ├── dnl_r50-d8.py │ │ │ ├── dpt_vit-b16.py │ │ │ ├── emanet_r50-d8.py │ │ │ ├── encnet_r50-d8.py │ │ │ ├── erfnet_fcn.py │ │ │ ├── fast_scnn.py │ │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ │ ├── fcn_hr18.py │ │ │ ├── fcn_r50-d8.py │ │ │ ├── fcn_unet_s5-d16.py │ │ │ ├── fpn_poolformer_s12.py │ │ │ ├── fpn_r50.py │ │ │ ├── gcnet_r50-d8.py │ │ │ ├── icnet_r50-d8.py │ │ │ ├── isanet_r50-d8.py │ │ │ ├── lraspp_m-v3-d8.py │ │ │ ├── nonlocal_r50-d8.py │ │ │ ├── ocrnet_hr18.py │ │ │ ├── ocrnet_r50-d8.py │ │ │ ├── pointrend_r50.py │ │ │ ├── psanet_r50-d8.py │ │ │ ├── pspnet_r50-d8.py │ │ │ ├── pspnet_unet_s5-d16.py │ │ │ ├── san_vit-b16.py │ │ │ ├── segformer_mit-b0.py │ │ │ ├── segmenter_vit-b16_mask.py │ │ │ ├── setr_mla.py │ │ │ ├── setr_naive.py │ │ │ ├── setr_pup.py │ │ │ ├── stdc.py │ │ │ ├── twins_pcpvt-s_fpn.py │ │ │ ├── twins_pcpvt-s_upernet.py │ │ │ ├── upernet_beit.py │ │ │ ├── upernet_convnext.py │ │ │ ├── upernet_mae.py │ │ │ ├── upernet_r50.py │ │ │ ├── upernet_swin.py │ │ │ ├── upernet_vit-b16_ln_mln.py │ │ │ └── vpd_sd.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_240k.py │ │ │ ├── schedule_25k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ ├── swin │ │ ├── README.md │ │ ├── metafile.yaml │ │ ├── swin-base-patch4-window12-in1k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-base-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-base-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-base-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-large-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-large-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-small-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ ├── swin-tiny-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py │ │ └── swin-tiny-patch4-window7_upernet_1xb8-20k_levir-256x256.py │ ├── upernet_sparx_mamba_b_ade20k.py │ ├── upernet_sparx_mamba_s_ade20k.py │ └── upernet_sparx_mamba_t_ade20k.py ├── models │ ├── __init__.py │ ├── csm_triton.py │ └── sparx_mamba.py ├── readme.md ├── scripts │ ├── dist_test.sh │ └── dist_train.sh ├── test.py └── train.py ├── train.py └── validate.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/LICENSE -------------------------------------------------------------------------------- /detection/config/_base_/datasets/ade20k_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/ade20k_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/ade20k_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/ade20k_panoptic.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/ade20k_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/ade20k_semantic.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_caption.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/coco_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/coco_semantic.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/dsdl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/dsdl.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/isaid_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/isaid_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/mot_challenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/mot_challenge.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/mot_challenge_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/mot_challenge_det.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/mot_challenge_reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/mot_challenge_reid.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/objects365v1_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/objects365v1_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/objects365v2_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/objects365v2_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/openimages_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/openimages_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/refcoco+.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/refcoco+.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/refcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/refcoco.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/refcocog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/refcocog.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/semi_coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/semi_coco_detection.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/v3det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/v3det.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /detection/config/_base_/datasets/youtube_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/datasets/youtube_vis.py -------------------------------------------------------------------------------- /detection/config/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/config/_base_/models/cascade-mask-rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/cascade-mask-rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/cascade-rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/cascade-rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/fast-rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/fast-rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/faster-rcnn_r50-caffe-c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/faster-rcnn_r50-caffe-c4.py -------------------------------------------------------------------------------- /detection/config/_base_/models/faster-rcnn_r50-caffe-dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/faster-rcnn_r50-caffe-dc5.py -------------------------------------------------------------------------------- /detection/config/_base_/models/faster-rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/faster-rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/mask-rcnn_r50-caffe-c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/mask-rcnn_r50-caffe-c4.py -------------------------------------------------------------------------------- /detection/config/_base_/models/mask-rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/mask-rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/rpn_r50-caffe-c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/rpn_r50-caffe-c4.py -------------------------------------------------------------------------------- /detection/config/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /detection/config/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/models/ssd300.py -------------------------------------------------------------------------------- /detection/config/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/config/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/config/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/config/mask-rcnn_swin-t-p4-w7_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/mask-rcnn_swin-t-p4-w7_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_b_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_b_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_b_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_b_fpn_3x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_s_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_s_fpn_3x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_t_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_t_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/config/maskrcnn_sparx_mamba_t_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/config/maskrcnn_sparx_mamba_t_fpn_3x_coco.py -------------------------------------------------------------------------------- /detection/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/__init__.py -------------------------------------------------------------------------------- /detection/models/__pycache__/__init__.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/__pycache__/__init__.cpython-311.pyc -------------------------------------------------------------------------------- /detection/models/__pycache__/csm_triton.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/__pycache__/csm_triton.cpython-311.pyc -------------------------------------------------------------------------------- /detection/models/__pycache__/sparx_mamba.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/__pycache__/sparx_mamba.cpython-311.pyc -------------------------------------------------------------------------------- /detection/models/csm_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/csm_triton.py -------------------------------------------------------------------------------- /detection/models/sparx_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/models/sparx_mamba.py -------------------------------------------------------------------------------- /detection/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/readme.md -------------------------------------------------------------------------------- /detection/scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/scripts/dist_test.sh -------------------------------------------------------------------------------- /detection/scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/scripts/dist_train.sh -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/detection/train.py -------------------------------------------------------------------------------- /images/arch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/images/arch.jpg -------------------------------------------------------------------------------- /kernels/selective_scan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/README.md -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cub_extra.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cub_extra.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cus/selective_scan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cus/selective_scan.cpp -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cus/selective_scan_bwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cus/selective_scan_bwd_kernel.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cus/selective_scan_core_bwd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cus/selective_scan_core_bwd.cu -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cus/selective_scan_core_fwd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cus/selective_scan_core_fwd.cu -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cus/selective_scan_fwd_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cus/selective_scan_fwd_kernel.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_bwd_kernel_oflex.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_bwd_kernel_oflex.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_core_bwd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_core_bwd.cu -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_core_fwd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_core_fwd.cu -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_fwd_kernel_oflex.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_fwd_kernel_oflex.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_oflex.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_oflex.cpp -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_oflex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_oflex.h -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/reverse_scan.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/reverse_scan.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/selective_scan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/selective_scan.h -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/selective_scan_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/selective_scan_common.h -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/static_switch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/static_switch.h -------------------------------------------------------------------------------- /kernels/selective_scan/csrc/selective_scan/uninitialized_copy.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/csrc/selective_scan/uninitialized_copy.cuh -------------------------------------------------------------------------------- /kernels/selective_scan/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/setup.py -------------------------------------------------------------------------------- /kernels/selective_scan/test_selective_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/test_selective_scan.py -------------------------------------------------------------------------------- /kernels/selective_scan/test_selective_scan_easy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/test_selective_scan_easy.py -------------------------------------------------------------------------------- /kernels/selective_scan/test_selective_scan_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/kernels/selective_scan/test_selective_scan_speed.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/csm_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/models/csm_triton.py -------------------------------------------------------------------------------- /models/sparx_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/models/sparx_mamba.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/readme.md -------------------------------------------------------------------------------- /scripts/train_sparx_mamba_b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/scripts/train_sparx_mamba_b.sh -------------------------------------------------------------------------------- /scripts/train_sparx_mamba_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/scripts/train_sparx_mamba_s.sh -------------------------------------------------------------------------------- /scripts/train_sparx_mamba_t.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/scripts/train_sparx_mamba_t.sh -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/ann_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/ann_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/apcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/apcnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/bisenetv1_r18-d32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/bisenetv1_r18-d32.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/bisenetv2.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/ccnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/ccnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/cgnet.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/danet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/danet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/deeplabv3_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/deeplabv3_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/deeplabv3_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/deeplabv3_unet_s5-d16.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/deeplabv3plus_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/deeplabv3plus_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/dmnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/dmnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/dnl_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/dnl_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/dpt_vit-b16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/dpt_vit-b16.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/emanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/emanet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/encnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/encnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/erfnet_fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/erfnet_fcn.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fast_scnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fast_scnn.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fcn_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fcn_hr18.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fcn_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fcn_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fcn_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fcn_unet_s5-d16.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fpn_poolformer_s12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fpn_poolformer_s12.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/gcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/gcnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/icnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/icnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/isanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/isanet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/lraspp_m-v3-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/lraspp_m-v3-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/nonlocal_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/nonlocal_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/ocrnet_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/ocrnet_hr18.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/ocrnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/ocrnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/pointrend_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/pointrend_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/psanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/psanet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/pspnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/pspnet_r50-d8.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/pspnet_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/pspnet_unet_s5-d16.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/san_vit-b16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/san_vit-b16.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/segformer_mit-b0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/segformer_mit-b0.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/segmenter_vit-b16_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/segmenter_vit-b16_mask.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/setr_mla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/setr_mla.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/setr_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/setr_naive.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/setr_pup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/setr_pup.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/stdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/stdc.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/twins_pcpvt-s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/twins_pcpvt-s_fpn.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/twins_pcpvt-s_upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/twins_pcpvt-s_upernet.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_beit.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_convnext.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_mae.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_swin.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_vit-b16_ln_mln.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/upernet_vit-b16_ln_mln.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/vpd_sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/models/vpd_sd.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_240k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_240k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_25k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_25k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /segmentation/configs/swin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/README.md -------------------------------------------------------------------------------- /segmentation/configs/swin/metafile.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/metafile.yaml -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-base-patch4-window12-in1k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-base-patch4-window12-in1k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-base-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-base-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-base-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-base-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-base-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-base-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-large-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-large-patch4-window12-in22k-384x384-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-large-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-large-patch4-window7-in22k-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-small-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-small-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-tiny-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-tiny-patch4-window7-in1k-pre_upernet_8xb2-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /segmentation/configs/swin/swin-tiny-patch4-window7_upernet_1xb8-20k_levir-256x256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/swin/swin-tiny-patch4-window7_upernet_1xb8-20k_levir-256x256.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_sparx_mamba_b_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/upernet_sparx_mamba_b_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_sparx_mamba_s_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/upernet_sparx_mamba_s_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_sparx_mamba_t_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/configs/upernet_sparx_mamba_t_ade20k.py -------------------------------------------------------------------------------- /segmentation/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .sparx_mamba import * -------------------------------------------------------------------------------- /segmentation/models/csm_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/models/csm_triton.py -------------------------------------------------------------------------------- /segmentation/models/sparx_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/models/sparx_mamba.py -------------------------------------------------------------------------------- /segmentation/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/readme.md -------------------------------------------------------------------------------- /segmentation/scripts/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/scripts/dist_test.sh -------------------------------------------------------------------------------- /segmentation/scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/scripts/dist_train.sh -------------------------------------------------------------------------------- /segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/test.py -------------------------------------------------------------------------------- /segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/segmentation/train.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/train.py -------------------------------------------------------------------------------- /validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LMMMEng/SparX/HEAD/validate.py --------------------------------------------------------------------------------