HZAI-ZJNU / Mamba-YOLO

the official pytorch implementation of “Mamba-YOLO：SSMs-based for Object Detection”

HZAI-ZJNU/Mamba-YOLO Issues

Could you tell me how to use the yolo-manba-seg.yaml?
Updated a month ago14
What are the extra kernels in selective scan used for?
Closed a month ago
论文图三的一点bug
Updated a month ago1
AttributeError: 'YOLO' object has no attribute 'test'
Updated a month ago4
RuntimeError: CUDA error: no kernel image is available for execution on the device Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
Updated a month ago1
Please provide example code for inference model
Updated a month ago1
RuntimeError: No CUDA GPUs are available
Updated a month ago2
Invalid CUDA 'device=0,1,2,3,4,5,6,7' requested. Use 'device=cpu' or pass valid CUDA device(s) if available, i.e. 'device=0' or 'device=0,1,2,3' for Multi-GPU.
Updated a month ago1
cd selective_scan &&pip install . 报错
Updated a month ago7
Expected u.is_cuda() to be true
Updated a month ago2
为什么运行起来有很多pt_main_thread子进程？
Updated a month ago
RuntimeError: Expected u.is_cuda() to be true, but got false. (Could this error message be improved? If so, please report an enhancement request to PyTorch.) During handling of the above exception, another exception occurred: RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
Updated a month ago
安装conda，torch的版本详细信息能否提供一下呢？试了很多都没有成功
Updated a month ago1
Mamba-yolo performs even worse than yolov8n on my dataset
Updated a month ago6
能否提供更详尽的关于mamba的环境依赖信息
Updated a month ago5
selective_scan安装报错
Updated a month ago3
Where is ultralytics/cfg/models/v8/mamba-yolo.yaml
Updated a month ago1
name 'selective_scan_cuda_core' is not defined
Closed a month ago
Reproducible results
Updated 2 months ago
train error
Updated 2 months ago4
Install selective_scan erro
Updated 2 months ago2
error about selective_scan_cuda_core
Updated 2 months ago2
Training command
Closed 2 months ago
A little problem about figure4 in paper
Updated 2 months ago1
selective_scan_cuda_core error with drop part errors
Updated 2 months ago
NameError: name 'selective_scan_cuda_core' is not defined. Did you mean: 'selective_scan_cuda'?
Updated 2 months ago1
Where the modifications regarding Mamba are located?
Closed 2 months ago2
Installation Guide
Closed 2 months ago
实验数据方面为什么只展示了coco val2017上的结果
Updated 2 months ago2
cd selective_scan pip install . 时候报错，visual studio 版本出错
Updated 2 months ago1
License Clarification
Updated 2 months ago
请问预训练权重会公开吗
Updated 2 months ago1
时延
Updated 2 months ago2