CV / STDC1
STDC1
reComputer R Series (CM5 + Hailo-8)
STDC1 实时语义分割,基于 reComputer R Series + Hailo-10H 部署。
53 次下载
大小
11 MB内存
4GB+精度
Hailo HEF / HailoRT选择你正在使用的设备,设置指南和文档将相应更新。
快速开始
部署
sudo docker run --rm \
--name cm5-hailo8-stdc1 \
--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/stdc1:latest \
python web_detection.py --model_path model/stdc1.hef --video_path video/test.mp4REST API
使用 REST API 进行推理。复制以下命令。
Curl
curl -X POST "http://<Board_IP>:8000/api/models/stdc1/predict" \
-F "file=@test.jpg"Python
import requests
resp = requests.post(
"http://<Board_IP>:8000/api/models/stdc1/predict",
files={"file": open("test.jpg", "rb")},
timeout=30,
)
print(resp.json())模型详情
reComputer R Series (CM5 + Hailo-8)STDC1 在 reComputer R Series(CM5 + Hailo-8)上运行
STDC1(Short-Term Dense Concatenate)通过 HailoRT 在 Hailo-8 上执行实时语义分割,支持 19 个 Cityscapes 类别。
本文适用于配备 PCIe Hailo-8 加速器的 reComputer R Series(CM5 + Hailo-8)。
模型信息
| 属性 | 值 |
|---|---|
| 架构 | STDC1(Short-Term Dense Concatenate) |
| 任务 | 语义分割 |
| Input | 1024x1920x3 RGB |
| Output | 1024x1920 class mask (19 classes) |
| 参数量 | 8.27M |
| mIoU (Cityscapes) | 73.8% hardware reference |
| HEF | Hailo Model Zoo v2.19.0, Hailo-8 |
上述精度来自 Hailo Model Zoo 参考数据,并非在 CM5 上实测的基准结果。
Cityscapes 类别
| ID | 类别 | ID | 类别 | ID | 类别 |
|---|---|---|---|---|---|
| 0 | 道路 | 7 | 交通标志 | 14 | 卡车 |
| 1 | 人行道 | 8 | 植被 | 15 | 公交车 |
| 2 | 建筑 | 9 | 地形 | 16 | 火车 |
| 3 | 墙 | 10 | 天空 | 17 | 摩托车 |
| 4 | 围栏 | 11 | 行人 | 18 | 自行车 |
| 5 | 杆 | 12 | 骑行者 | ||
| 6 | 交通信号灯 | 13 | 汽车 |
硬件与主机设置
| 项目 | 值 |
|---|---|
| 开发板 | 配备 Raspberry Pi CM5 的 reComputer R Series |
| 加速器 | Hailo-8 通过 PCIe 连接,设备为 /dev/hailo0 |
| 运行时 | HailoRT 4.23.x |
| 容器内 Python | 3.11,aarch64 |
启动容器前,请在主机上安装 Hailo-8 软件包,并确认驱动和设备已就绪:
bash
sudo apt update
sudo apt install hailo-all
sudo reboot
# 重启后
hailortcli --version
hailortcli fw-control identify
ls -l /dev/hailo0此镜像要求使用 HailoRT 4.23.x。
使用演示视频运行
bash
sudo docker run --rm \
--name cm5-hailo8-stdc1 \
--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/stdc1:latest \
python web_detection.py --model_path model/stdc1.hef --video_path video/test.mp4打开 http://<Board_IP>:8000 查看网页预览。
USB 摄像头模式
bash
sudo docker run --rm \
--name cm5-hailo8-stdc1 \
--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/stdc1:latest \
python web_detection.py --model_path model/stdc1.hef --camera_id 0REST API
预测端点:
text
POST http://<Board_IP>:8000/api/models/stdc1/predict图片请求示例:
bash
curl -X POST "http://<Board_IP>:8000/api/models/stdc1/predict" \
-F "file=@test.jpg"| 端点 | 方法 | 用途 |
|---|---|---|
/ | GET | 网页预览界面 |
/api/models/stdc1/predict | POST | 分割掩码(JSON) |
/api/models/stdc1/visualize | POST | 叠加图像(JPEG) |
/api/models/stdc1/classes | GET | Cityscapes 类别列表 |
/api/video_feed | GET | MJPEG 预览流 |
开发说明
- 源码模块:
src/rpi5_hailo8_stdc1/ - Dockerfile:
docker/hailo8/stdc1.dockerfile - 容器镜像:
ghcr.io/seeed-projects/recomputer-hailo8-cv/stdc1:latest - 该模块使用 HailoRT 4.23 VStreams API 和 Hailo-8 专用 HEF。
输入与输出
输入:图像(1024×1920 RGB)。输出:19 类语义分割掩码(Cityscapes)。