Mamba YOLO环境配置和测试（Ubuntu20.04-cuda11.8）

Title：Mamba-YOLO：SSMs-based for Object Detection

code：https://github.com/HZAI-ZJNU/Mamba-YOLO.git

论文地址：https://arxiv.org/abs/2406.05835

一、下载代码

git clone https://github.com/HZAI-ZJNU/Mamba-YOLO.git

二、环境配置

我是用之前自己的环境，也可以按照官方Readme要求进行配置，但

cd selective_scan && pip install . && cd ..

报错：

ERROR: Could not build wheels for selective_scan, which is required to install pyproject.toml-based projects

这里直接从这个博客下载了selective-scan.whl (CUDA 11.8 包含selective_scan_core），并安装。

三、测试

测试代码来自：Predict - Ultralytics YOLO Docs

from PIL import Image

from ultralytics import YOLO

# Load a pretrained YOLOv8n model
model = YOLO("yolov8n.pt")

# Run inference on 'bus.jpg'
results = model(["bus.jpg", "zidane.jpg"])  # results list

# Visualize the results
for i, r in enumerate(results):
    # Plot results image
    im_bgr = r.plot()  # BGR-order numpy array
    im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image

    # Show results to screen (in supported environments)
    r.show()

    # Save results to disk
    r.save(filename=f"results{i}.jpg")

测试结果：

注：如果不安装 selective_scan_core，也可以进行测试，但会出现这个错误！！！结果如下：