CV / SSD
SSD MobileNet V1
reComputer R Series (CM5 + Hailo-8)
Hailo-8 搭載の reComputer R Series で、SSD MobileNet V1 による COCO 80 クラスの物体検出を実行します(オンチップ NMS)。
使用しているデバイスを選択してください。セットアップガイドとドキュメントがそれに応じて更新されます。
はじめる
sudo docker run --rm \
--name cm5-hailo8-ssd-mobilenet-v1 \
--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/ssd_mobilenet_v1:latest \
python web_detection.py --model_path model/ssd_mobilenet_v1.hef --video_path video/test.mp4REST API
REST API を使用して推論を実行します。以下のコマンドをコピーしてください。
curl -X POST "http://<Board_IP>:8000/api/models/ssd_mobilenet_v1/predict" \
-F "file=@test.jpg"import requests
response = requests.post(
"http://<Board_IP>:8000/api/models/ssd_mobilenet_v1/predict",
files={"file": open("test.jpg", "rb")},
timeout=30,
)
print(response.json())モデル詳細
reComputer R Series (CM5 + Hailo-8) で SSD MobileNet V1 を実行
SSD MobileNet V1 は MobileNet V1 バックボーンを持つ従来の TensorFlow SSD 検出器で、Hailo-8 上で COCO 80 クラスを検出します。定義済みアンカーを用いた NMS はチップ上で実行され(Hailo HPP、meta_arch=ssd)、ホストは NMS 後のテンソルのみを解析します。
このページは、PCIe Hailo-8 アクセラレータを搭載した reComputer R Series (CM5 + Hailo-8) を対象とします。
モデル情報
| 項目 | 値 |
|---|---|
| アーキテクチャ | SSD + MobileNet V1 |
| タスク | 物体検出 |
| 入力 | 300x300x3 RGB (normalize_in_net mean=127.5/std=127.5) |
| 出力 | チップ上 NMS テンソル、NMS 後の形状 90x8x1 |
| クラス数 | 90 スロット(labels_offset=1 で COCO ID 1..90 に対応、10 個未使用) |
| パラメータ数 | 6.79M |
| 演算量 | 2.5G |
| 後処理 | チップ上 NMS(HPP)、ホストは NMS 後のテンソルを解析 |
| HEF | Hailo Model Zoo v2.19.0, Hailo-8 |
精度は Hailo Model Zoo の参照値であり、CM5 上で測定したベンチマークではありません。
ハードウェアとホストの設定
| 項目 | 値 |
|---|---|
| ボード | Raspberry Pi CM5 搭載 reComputer R Series |
| アクセラレータ | PCIe 接続 Hailo-8、デバイスは /dev/hailo0 |
| ランタイム | HailoRT 4.23.x |
| コンテナ内 Python | 3.11, aarch64 |
ホストに Hailo-8 パッケージをインストールし、コンテナ起動前にドライバーとデバイスの準備を確認してください:
sudo apt update
sudo apt install hailort hailort-pcie-driver python3-hailort
sudo reboot
# After reboot
hailortcli --version
hailortcli fw-control identify
ls -l /dev/hailo0このイメージが想定する HailoRT バージョンは 4.23.x です。
デモ動画で実行
sudo docker run --rm \
--name cm5-hailo8-ssd-mobilenet-v1 \
--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/ssd_mobilenet_v1:latest \
python web_detection.py --model_path model/ssd_mobilenet_v1.hef --video_path video/test.mp4http://<Board_IP>:8000 を開いて Web プレビューを確認します。
USB カメラモード
sudo docker run --rm \
--name cm5-hailo8-ssd-mobilenet-v1 \
--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/ssd_mobilenet_v1:latest \
python web_detection.py --model_path model/ssd_mobilenet_v1.hef --camera_id 0REST API
予測エンドポイント:
POST http://<Board_IP>:8000/api/models/ssd_mobilenet_v1/predict画像リクエストの例:
curl -X POST "http://<Board_IP>:8000/api/models/ssd_mobilenet_v1/predict" \
-F "file=@test.jpg"| エンドポイント | メソッド | 用途 |
|---|---|---|
/ | GET | Web プレビュー画面 |
/api/models/ssd_mobilenet_v1/predict | POST | 検出結果(JSON) |
/api/video_feed | GET | MJPEG プレビューストリーム |
実装上の注意
- HEF は NMS をチップ上で実行します(
device_pre_post_layers: nms=true)。アプリは公式tf_postproc_nmsに従って NMS 後のテンソルのみを解析するため、nms_threshは無視されます(API 互換性のみ)。 - NMS 後の行は
[ymin, xmin, ymax, xmax, score, ...]で、300x300 の letterbox 入力に対して [0,1] に正規化されます。アプリがピクセル座標への変換と letterbox の補正を行います。 normalize_in_netは mean=127.5/std=127.5(従来の SSD 正規化(pixel-127.5)/127.5)です。アプリは letterbox 後に生の uint8 RGB ピクセルを入力し、手動正規化は行いません。- HailoRT は NMS vstream をクラスごとの可変長リスト(NMS-by-score)で返します。パーサーは object/dense 形式にも対応します。初回推論で生の型と形状を記録し、実機検証を可能にします。
- クラス対応:
cls_id(0..89)→ COCO カテゴリ IDcls_id+1(labels_offset=1)。未使用の 10 個の COCO ID は「N/A」とし、描画しません。 - 既定値:信頼度 0.25、IOU 0.45(NMS はチップ上のため IOU は無効)。Model Zoo 評価は 0.3 / 0.6 を使用します。信頼度スライダーを下げると検出を多く確認できます。
開発上の注意
- ソースモジュール:
src/rpi5_hailo8_ssd_mobilenet_v1/ - Dockerfile:
docker/hailo8/ssd_mobilenet_v1.dockerfile - コンテナ:
ghcr.io/seeed-projects/recomputer-hailo8-cv/ssd_mobilenet_v1:latest - モジュールは HailoRT 4.23 VStreams API と Hailo-8 専用 HEF を使用します。
- ファミリー:ssd(バリアント:ssd_mobilenet_v1 / ssd_mobilenet_v2 は計画中)。これは V1 ビルドです。
入力と出力
Input: image, video, or USB camera frame. Output: COCO 80-class detection boxes with confidences, plus an annotated MJPEG preview.