CV / SCRFD

SCRFD 2.5G

SCRFD 2.5Gは、reComputer Rシリーズ上での顔検出において速度と精度のバランスを取ります。

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

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

はじめる

デプロイ
sudo docker run --rm \
  --name cm5-hailo-scrfd-2-5g \
  --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/scrfd:latest \
  python web_detection.py --model_path model/scrfd_2.5g.hef --video_path video/test.mp4

REST API

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

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

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

モデル詳細

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

Hailo-8アクセラレータ上でのバウンディングボックス、信頼度スコア、5点ランドマーク付き顔検出。

このページはreComputer-R20-CV内のrpi5_hailo8_scrfd/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/scrfd:latest \
  python web_detection.py --model_path model/scrfd_500m.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/scrfd:latest \
  python web_detection.py --model_path model/scrfd_500m.hef --camera_id 0

HEFバリアントの切り替え

別の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/scrfd:latest \
  python web_detection.py --model_path model/scrfd_2.5g.hef --video_path video/test.mp4

HEFファイル

HEFサイズ備考
scrfd_500m.hef1.00 MBデフォルトSCRFDモデル
scrfd_2.5g.hef2.96 MBバランス型SCRFDモデル
scrfd_10g.hef13.56 MB大規模SCRFDモデル

REST API

予測エンドポイント:

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

画像リクエストの例:

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

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

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

入力と出力

入力: 画像、動画、またはUSBカメラフレーム。出力: 顔バウンディングボックス、スコア、5つのランドマーク、および注釈付きMJPEGプレビュー。