CV / NanoDet

NanoDet-RepVGG-a1-640

reComputer R Series (CM5 + Hailo-8)

COCO 80 クラスの物体検出。nanodet_repvgg_a1_640(オンチップ NMS)。reComputer R Series + Hailo-8 または Hailo-10H 対応。

70 ダウンロード
サイズ
9.8 MB
メモリ
4GB+
精度
Hailo HEF / HailoRT

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

はじめる

デプロイ
sudo docker run --rm \
  --name cm5-hailo8-nanodet-repvgg-a1-640 \
  --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/nanodet_repvgg_a1_640:latest \
  python web_detection.py --model_path model/nanodet_repvgg_a1_640.hef --video_path video/test.mp4

REST API

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

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

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

モデル詳細

reComputer R Series (CM5 + Hailo-8)

reComputer R Series (CM5 + Hailo-8) で NanoDet-RepVGG-a1-640 を実行

nanodet_repvgg_a1_640 は HailoRT を介して Hailo-8 上で COCO 80 クラスの物体検出を行います。nanodet_repvgg と同じベース(base/nanodet.yaml)ですが、より大きい RepVGG-A1 バックボーンと 640x640 入力で精度を高めます。NMS はチップ上で実行され(Hailo HPP、meta_arch=yolov8)、ホストは NMS 後のテンソルのみを解析します。

このページは、PCIe Hailo-8 アクセラレータを搭載した reComputer R Series (CM5 + Hailo-8) を対象とします。

モデル情報

項目値
アーキテクチャNanoDet + RepVGG-A1 (base/nanodet.yaml)
タスク物体検出
入力640x640x3 BGR (input_conversion bgr_to_rgb + normalize_in_net)
出力チップ上 NMS テンソル、NMS 後の形状 80x5x100
クラス数80(COCO、meta_arch=yolov8 により 0 始まり)
パラメータ数10.79M
演算量42.8G
後処理チップ上 NMS(HPP)、ホストは NMS 後のテンソルを解析
HEFHailo 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
コンテナ内 Python3.11, aarch64

ホストに Hailo-8 パッケージをインストールし、コンテナ起動前にドライバーとデバイスの準備を確認してください:

bash
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 です。

デモ動画で実行

bash
sudo docker run --rm \
  --name cm5-hailo8-nanodet-repvgg-a1-640 \
  --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/nanodet_repvgg_a1_640:latest \
  python web_detection.py --model_path model/nanodet_repvgg_a1_640.hef --video_path video/test.mp4

http://<Board_IP>:8000 を開いて Web プレビューを確認します。

USB カメラモード

bash
sudo docker run --rm \
  --name cm5-hailo8-nanodet-repvgg-a1-640 \
  --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/nanodet_repvgg_a1_640:latest \
  python web_detection.py --model_path model/nanodet_repvgg_a1_640.hef --camera_id 0

REST API

予測エンドポイント:

text
POST http://<Board_IP>:8000/api/models/nanodet_repvgg_a1_640/predict

画像リクエストの例:

bash
curl -X POST "http://<Board_IP>:8000/api/models/nanodet_repvgg_a1_640/predict" \
  -F "file=@test.jpg"
エンドポイントメソッド用途
/GETWeb プレビュー画面
/api/models/nanodet_repvgg_a1_640/predictPOST検出結果(JSON)
/api/video_feedGETMJPEG プレビューストリーム

実装上の注意

  • ベース(base/nanodet.yaml)とチップ上 NMS(HPP)後処理は nanodet_repvgg と同じです。バックボーン(RepVGG-A1)と入力サイズ(640x640)のみが異なります。
  • HEF は NMS をチップ上で実行します(device_pre_post_layers: nms=true, hpp=true、meta_arch=yolov8)。アプリは NMS 後のテンソルのみを解析するため、nms_thresh は無視されます(API 互換性のためだけに存在)。
  • NMS 後の行は [ymin, xmin, ymax, xmax, score] で、640x640 の letterbox 入力に対して [0,1] に正規化されます。アプリがピクセル座標への変換と letterbox の補正を行います。
  • normalize_in_net + チップ上の input_conversion(bgr_to_rgb):アプリは黒(0)で letterbox を行い、生の uint8 BGR ピクセルを入力します。手動正規化も cvtColor も行いません。
  • HailoRT は NMS vstream をクラスごとの可変長リスト(NMS-by-score)で返します。パーサーは object/dense 形式にも対応します。初回推論で生の型と形状を記録し、実機検証を可能にします。
  • クラス対応:cls_id(0..79)は標準 COCO 80 クラス一覧に直接対応します。
  • 既定値:信頼度 0.25、IOU 0.45(NMS はチップ上のため IOU は無効)。Model Zoo 評価は 0.05 / 0.6 を使用します。信頼度スライダーを下げると検出を多く確認できます。

開発上の注意

  • ソースモジュール: src/rpi5_hailo8_nanodet_repvgg_a1_640/
  • Dockerfile: docker/hailo8/nanodet_repvgg_a1_640.dockerfile
  • コンテナ: ghcr.io/seeed-projects/recomputer-hailo8-cv/nanodet_repvgg_a1_640:latest
  • モジュールは HailoRT 4.23 VStreams API と Hailo-8 専用 HEF を使用します。
  • ファミリー:nanodet(バリアント:nanodet_repvgg / nanodet_repvgg_a12 / nanodet_repvgg_a1_640)。これは nanodet_repvgg_a1_640 ビルドです。

入力と出力

Input: image, video, or USB camera frame. Output: COCO 80-class detection boxes with confidences, plus an annotated MJPEG preview.