CV / DAMO-YOLO

DAMO-YOLO

DAMO-YOLO(T/S/M 变体)在 reComputer R Series 上通过 Hailo-8 或 Hailo-10H 执行 COCO 80 类目标检测。

161 次下载
大小
8.8 MB
内存
4GB+
精度
Hailo HEF / HailoRT 4.23.x (Hailo-8) / 5.1.1 (Hailo-10H)

选择你正在使用的设备,设置指南和文档将相应更新。

快速开始

部署
sudo docker run --rm \
  --name cm5-hailo8-damoyolo-l20t \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.4.23.0:/usr/lib/libhailort.so.4.23.0:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l20_t:latest \
  python web_detection.py --model_path model/damoyolo_tinynas_l20_t.hef --video_path video/test.mp4

模型详情

reComputer R Series (CM5 + Hailo-8)

DAMO-YOLO 在 reComputer R Series(CM5 + Hailo-8)上的部署

DAMO-YOLO(TinyNAS 主干网络)通过 HailoRT 在 Hailo-8 上执行 COCO 80 类目标检测,使用 nanodet_split 检测头和 DFL 框回归。一张卡片,三个变体,通过 Size 下拉框选择。

本页面向搭配 PCIe Hailo-8 加速器的 reComputer R Series(CM5 + Hailo-8)。

模型信息

变体参数量计算量mAP(COCO)打包 HEF
T (tinynasL20)11.35M18.0G42.88.8 MB
S (tinynasL25)16.25M37.6G46.514 MB
M (tinynasL35)33.98M61.6G49.727 MB
属性值
任务目标检测
输入640x640x3 RGB(归一化为空操作;原始 uint8)
输出6 个头:box(80x80x68、40x40x68、20x20x68)+ cls(80x80x81、40x40x81、20x20x81)
类别80(COCO)
后处理nanodet_split(DFL,regression_length=16),仅片上 sigmoid
HEFHailo Model Zoo v2.19.0,Hailo-8

精度数值为 Hailo Model Zoo 的参考值,不是在 CM5 上实测的基准。

硬件与主机设置

项目值
开发板reComputer R Series(树莓派 CM5)
加速器Hailo-8,PCIe 接口,设备节点为 /dev/hailo0
运行时HailoRT 4.23.x
容器内 Python3.11,aarch64
bash
sudo apt update
sudo apt install hailo-all
sudo reboot

# After reboot
hailortcli fw-control identify
ls /dev/hailo0

# Docker
curl -fsSL https://get.docker.com -o get-docker.sh
sudo sh get-docker.sh --mirror Aliyun
sudo systemctl enable docker
sudo systemctl start docker

使用演示视频运行

请选择与变体对应的命令。

DAMO-YOLO T (tinynasL20)

bash
sudo docker run --rm \
  --name cm5-hailo8-damoyolo-l20t \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.4.23.0:/usr/lib/libhailort.so.4.23.0:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l20_t:latest \
  python web_detection.py --model_path model/damoyolo_tinynas_l20_t.hef --video_path video/test.mp4

DAMO-YOLO S (tinynasL25)

bash
sudo docker run --rm \
  --name cm5-hailo8-damoyolo-l25s \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.4.23.0:/usr/lib/libhailort.so.4.23.0:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l25_s:latest \
  python web_detection.py --model_path model/damoyolo_tinynas_l25_s.hef --video_path video/test.mp4

DAMO-YOLO M (tinynasL35)

bash
sudo docker run --rm \
  --name cm5-hailo8-damoyolo-l35m \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.4.23.0:/usr/lib/libhailort.so.4.23.0:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l35_m:latest \
  python web_detection.py --model_path model/damoyolo_tinynas_l35_m.hef --video_path video/test.mp4

打开 http://<Board_IP>:8000 查看网页预览。

USB 摄像头模式(以 S 变体为例)

bash
sudo docker run --rm \
  --name cm5-hailo8-damoyolo-l25s \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  --device /dev/video0:/dev/video0 \
  -v /usr/lib/libhailort.so.4.23.0:/usr/lib/libhailort.so.4.23.0:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l25_s:latest \
  python web_detection.py --model_path model/damoyolo_tinynas_l25_s.hef --camera_id 0

REST API

bash
curl -X POST "http://<Board_IP>:8000/api/models/damoyolo_tinynas_l25_s/predict" \
  -F "file=@test.jpg"
端点方法用途
/api/models/damoyolo_tinynas_<variant>/predictPOST检测结果(JSON)
/api/video_feedGETMJPEG 预览
/api/configGET / POST置信度阈值

开发说明

  • 源码模块:src/rpi5_hailo8_damoyolo_tinynas_l20_t/、..._l25_s/、..._l35_m/
  • Dockerfile:docker/hailo8/damoyolo_tinynas_l*.dockerfile
  • 容器镜像:ghcr.io/seeed-projects/recomputer-hailo8-cv/damoyolo_tinynas_l{20_t,25_s,35_m}:latest
  • 三个变体使用相同的 nanodet_split CPU 解码——仅主干网络(TinyNAS-L20/L25/L35)不同。
  • HEF:Hailo Model Zoo v2.19.0 的 hailo8 构建。

输入与输出

输入:图像、视频或 USB 摄像头帧。输出:带置信度的 COCO 80 类检测框和带标注的 MJPEG 预览。