CV / YOLOv8

YOLOv8m_pose

YOLOv8m Poseは、R20 HailoRTデモ用のより大規模な人物キーポイント推定HEFを提供します。

サイズ
30.14 MB
メモリ
4GB+
精度
Hailo HEF / HailoRT 4.23.x

使用しているデバイスを選択してください。セットアップガイドとドキュメントがそれに応じて更新されます。

はじめる

デプロイ
sudo docker run --rm \
  --name cm5-hailo-yolov8m-pose \
  --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-r20-cv/yolov8_pose:latest \
  python web_detection.py --model_path model/yolov8m_pose.hef --video_path video/test.mp4

REST API

REST API を使用して推論を実行します。以下のコマンドをコピーしてください。

Curl
curl -X POST "http://<R20_IP>:8000/api/models/yolov8_pose/predict" \
  -F "file=@test.jpg"
Python
import requests

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

モデル詳細

YOLOv8 Pose(reComputer Rシリーズ CM5 + Hailo対応)

YOLOv8 PoseはHailoRT FastAPIデモを通じて人物バウンディングボックスとキーポイントを返します(reComputer Rシリーズ対応)。

このページはreComputer-R20-CV内のrpi5_hailo8_yolov8_pose/README.mdから改変されています。対象はPCIe Hailo-8アクセラレータ搭載の**reComputer Rシリーズ(CM5 + Hailo)**です。

ハードウェアとホストのセットアップ

項目
ボードRaspberry Pi CM5搭載reComputer Rシリーズ
アクセラレータHailo-8 M.2(PCIe接続)、/dev/hailo0として公開
OSRaspberry Pi OS Bookworm, aarch64
ホストドライバhailo-all aptパッケージ
検証済みHailoRT4.23.x; ホストドライバ、ファームウェア、ネイティブライブラリ、Python wheelのメジャー・マイナーバージョンが一致している必要があります
bash
sudo apt update
sudo apt install hailo-all
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.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-r20-cv/yolov8_pose:latest \
  python web_detection.py --model_path model/yolov8s_pose.hef --video_path video/test.mp4

http://<R20_IP>:8000を開いてWebプレビューを表示します。イメージにはモジュールソース、デモ動画、HailoRT Python wheel、および以下に示すHEFファイルが含まれています。

libhailort.soバインドマウントは意図的にホストから取得されます。ファームウェアが4.23.0でない場合は、両方のパスをhailortcli fw-control identifyで報告されたバージョンに置き換えてください。メジャー・マイナーバージョンが異なる場合はイメージを再ビルドしてください。

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.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-r20-cv/yolov8_pose:latest \
  python web_detection.py --model_path model/yolov8s_pose.hef --camera_id 0

HEFバリアントの切り替え

別の--model_pathを渡してイメージのデフォルトを上書きします:

bash
sudo docker run --rm --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-r20-cv/yolov8_pose:latest \
  python web_detection.py --model_path model/yolov8m_pose.hef --video_path video/test.mp4

HEFファイル

HEFサイズ備考
yolov8s_pose.hef21.47 MBデフォルト姿勢モデル
yolov8m_pose.hef49.34 MB大規模姿勢モデル

REST API

予測エンドポイント:

text
POST http://<R20_IP>:8000/api/models/yolov8_pose/predict

画像リクエストの例:

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

共通サービスエンドポイント:

エンドポイントメソッド目的
/GETWebプレビューUI
/api/models/yolov8_pose/predictPOSTアップロードされた画像/動画または現在のフレームからの単一フレーム推論
/api/video_feedGETMJPEGプレビューストリーム
/api/configGET / POSTランタイムしきい値の読み取りまたは更新
/api/video/uploadPOSTオフライン分析用の動画をアップロード
/api/video/analyzePOSTオフライン分析を開始
/api/video/statusGET分析の進行状況をポーリング
/api/video/listGETアップロードおよび処理済み動画の一覧表示
/api/video/download/{filename}GET処理済み出力をダウンロード

開発ノート

  • ソースモジュール: src/rpi5_hailo8_yolov8_pose/
  • Dockerfile: docker/hailo8/yolov8_pose.dockerfile
  • メインサービス: web_detection.pyはHailoRT推論、後処理、MJPEGプレビュー、REST予測、オフライン動画分析をラップします。
  • カスタムHEFに差し替える場合は、出力レイアウトをモジュールの後処理コードと互換性のあるものにするか、post_process_hailo()を更新してください。

入力と出力

入力: 画像、動画、またはUSBカメラフレーム。出力: 人物ボックス、信頼度、キーポイント、および注釈付きMJPEGプレビュー。