CV / MSPN 姿态估计

MSPN RegNetX-800MF

reComputer R Series (CM5 + Hailo-10H)

MSPN 单人姿态估计，支持 17 个 COCO 关键点，基于 reComputer R Series + Hailo-8 或 Hailo-10H 部署。

192 次下载

大小

5.8 MB

内存

4GB+

精度

Hailo HEF / HailoRT

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

平台

推理引擎

快速开始

部署

sudo docker run --rm \
  --name cm5-hailo-mspn-pose \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/mspn_regnetx_800mf:latest \
  python web_detection.py --model_path model/mspn_regnetx_800mf.hef --video_path video/test.mp4

REST API

使用 REST API 进行推理。复制以下命令。

Curl

curl -X POST "http://<Board_IP>:8000/api/models/mspn_regnetx_800mf/predict" \
  -F "file=@test.jpg"

Python

import requests

resp = requests.post(
    "http://<Board_IP>:8000/api/models/mspn_regnetx_800mf/predict",
    files={"file": open("test.jpg", "rb")},
    timeout=30,
)
print(resp.json())

模型详情

reComputer R Series

MSPN RegNetX-800MF 在 reComputer R Series（CM5 + Hailo-10H）上运行

MSPN（Multi-Stage Pose Network）搭配 RegNetX-800MF 骨干网络，实现单人姿态估计，输出 17 个 COCO 关键点，通过 HailoRT 在 Hailo-10H 上运行。

本页面向搭载 PCIe Hailo-10H 加速器的 reComputer R Series（CM5 + Hailo-10H）。

模型信息

属性	值
架构	MSPN + RegNetX-800MF
任务	单人姿态估计
输入	256×192×3 RGB
输出	17 个关键点（COCO 格式）
参数量	7.17M
硬件 AP	69.8%
FPS	2034
来源	Hailo Model Zoo

COCO 关键点

ID	名称	ID	名称	ID	名称
0	nose	6	right_shoulder	13	left_knee
1	left_eye	7	left_elbow	14	right_knee
2	right_eye	8	right_elbow	15	left_ankle
3	left_ear	9	left_wrist	16	right_ankle
4	right_ear	10	right_wrist
5	left_shoulder	11	left_hip
		12	right_hip

硬件与主机设置

项目	值
开发板	reComputer R Series 搭配 Raspberry Pi CM5
加速器	Hailo-10H（PCIe 接口），以 `/dev/hailo0` 暴露
操作系统	Raspberry Pi OS Bookworm，aarch64
主机驱动	`hailo-h10-all` apt 包

bash

sudo apt update
sudo apt install hailo-h10-all -y
sudo 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

使用演示视频运行

bash

sudo docker run --rm --privileged --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/mspn_regnetx_800mf:latest \
  python web_detection.py --model_path model/mspn_regnetx_800mf.hef --video_path video/test.mp4

打开 http://<Board_IP>:8000 查看 Web 预览。镜像包含模块源码、演示视频、HailoRT Python wheel 以及下列 HEF 文件。

USB 摄像头模式

bash

sudo docker run --rm --privileged --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  --device /dev/video0:/dev/video0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/mspn_regnetx_800mf:latest \
  python web_detection.py --model_path model/mspn_regnetx_800mf.hef --camera_id 0

HEF 文件

HEF	大小	备注
`mspn_regnetx_800mf.hef`	5.8 MB	姿态估计模型

REST API

预测端点：

text

POST http://<Board_IP>:8000/api/models/mspn_regnetx_800mf/predict

示例图片请求：

bash

curl -X POST "http://<Board_IP>:8000/api/models/mspn_regnetx_800mf/predict" \
  -F "file=@test.jpg"

通用服务端点：

端点	方法	用途
`/`	GET	Web 预览界面
`/api/models/mspn_regnetx_800mf/predict`	POST	17 个 COCO 关键点（JSON）
`/api/video_feed`	GET	MJPEG 预览流

开发说明

源模块：src/hailo10h_mspn_regnetx_800mf/
Dockerfile：docker/hailo10h/mspn_regnetx_800mf.dockerfile
主服务：web_detection.py 封装了 HailoRT 推理、后处理、MJPEG 预览和 REST 预测。

输入与输出

输入：图像（256x192 RGB）。输出：17 个关键点，包含 x、y 和置信度（COCO 格式）。