CV / ResNet

ResNet50V2

ImageNet classification with configurable Top-K results, browser preview, and REST API access.

2 ダウンロード

精度

RKNN

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

プラットフォーム

サイズ

推論エンジン

はじめる

デプロイ

sudo docker run --rm --privileged --net=host \
  -e PYTHONUNBUFFERED=1 -e RKNN_LOG_LEVEL=0 \
  --device /dev/dri/renderD128:/dev/dri/renderD128 \
  -v /proc/device-tree/compatible:/proc/device-tree/compatible:ro \
  ghcr.io/seeed-projects/recomputer-rk-cv/rk3576-resnet50v2:latest \
  python3 web_classification.py --model_path model/rk3576_resnet50-v2-7.rknn --camera_id -1

REST API

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

Curl

curl http://localhost:8080/v1/chat/completions -d '{
  "model": "resnet50-v2-rknn",
  "messages": [{"role": "user", "content": "Hello"}]
}'

Python

import requests

resp = requests.post(
    "http://localhost:8080/v1/chat/completions",
    json={"model": "resnet50-v2-rknn", "messages": [{"role": "user", "content": "Hello"}]},
)
print(resp.json())

モデル詳細

reComputer RK3576およびRK3588でResNet50V2を実行する

このページはreComputer-RK-CVのResNet50V2 READMEを基にしています。Rockchip NPUでImageNet分類を実行し、ブラウザ、MJPEG、ヘルスチェック、RESTインターフェースを提供します。

モデル情報

プロパティ	値
タスク	画像分類
入力	`224 x 224` RGB画像
出力	設定可能なTop-K ImageNetラベルと信頼度スコア
RK3576モデル	`model/rk3576_resnet50-v2-7.rknn`
RK3588モデル	`model/rk3588_resnet50-v2-7.rknn`
ラベル	`model/synset.txt`

サービスを実行する

RK3576

bash

sudo docker run --rm --privileged --net=host \
  -e PYTHONUNBUFFERED=1 -e RKNN_LOG_LEVEL=0 \
  --device /dev/dri/renderD128:/dev/dri/renderD128 \
  -v /proc/device-tree/compatible:/proc/device-tree/compatible:ro \
  ghcr.io/seeed-projects/recomputer-rk-cv/rk3576-resnet50v2:latest \
  python3 web_classification.py \
  --model_path model/rk3576_resnet50-v2-7.rknn --camera_id -1

RK3588

bash

sudo docker run --rm --privileged --net=host \
  -e PYTHONUNBUFFERED=1 -e RKNN_LOG_LEVEL=0 \
  --device /dev/dri/renderD128:/dev/dri/renderD128 \
  -v /proc/device-tree/compatible:/proc/device-tree/compatible:ro \
  ghcr.io/seeed-projects/recomputer-rk-cv/rk3588-resnet50v2:latest \
  python3 web_classification.py \
  --model_path model/rk3588_resnet50-v2-7.rknn --camera_id -1

Webページはhttp://<BOARD_IP>:8000、OpenAPIはhttp://<BOARD_IP>:8000/docsで開きます。

入力モードと起動引数

アプリケーションは、アップロード画像、アップロードMP4から選択したフレーム、ループ再生するローカル動画、カメラフレームを受け付けます。

引数	デフォルト	説明
`--model_path`	必須	対象SoC向けに選択したRKNNモデル。
`--camera_id`	`1`	カメラインデックス。アップロード専用モードには`-1`を使用します。
`--video_path`	なし	ローカルMP4のパス。カメラより優先されます。
`--host`	`0.0.0.0`	FastAPIのリッスンアドレス。
`--port`	`8000`	サービスポート。

カメラNを選択する場合は/dev/videoNをコンテナへマッピングします。

REST API

エンドポイント： POST /api/models/resnet50v2/predict

Multipartフィールドはfile、video、timestamp、conf、topkです。

bash

curl -X POST "http://<BOARD_IP>:8000/api/models/resnet50v2/predict" \
  -F "file=@dog_224x224.jpg" -F "topk=5" -F "conf=0.01"

bash

curl -X POST "http://<BOARD_IP>:8000/api/models/resnet50v2/predict" \
  -F "video=@test.mp4" -F "timestamp=5.5" -F "topk=3"

レスポンス例：

json

{
  "success": true,
  "model": "resnet50v2",
  "source": "uploaded image",
  "predictions": [
    {"class": "n02099601 golden retriever", "confidence": 0.89}
  ],
  "image": {"width": 224, "height": 224}
}

GET /api/configはconf_threshを返します。POST /api/configを通じてJSONで更新します。リクエスト単位のconfが優先されます。

エンドポイント	用途
`GET /api/health`	サービス、プラットフォーム、モデル、準備状態。
`GET /api/video_feed`	注釈付きMJPEGプレビュー。
`POST /api/video/upload`	画像またはMP4をアップロードします。
`POST /api/video/analyze`	アップロード済みの`filename`を解析します。
`GET /api/video/status`	処理進捗とエラー。
`GET /api/video/list`	アップロード済みおよび生成済みファイル。
`GET /api/video/download/{filename}`	結果をダウンロードします。

モデルの置き換え

ランタイムは入力を224 x 224へリサイズし、BGRからRGBへ変換し、softmaxを適用してTop-K結果を絞り込みます。置き換えるモデルは想定される入力レイアウト、正規化、チャネル順、出力形式を維持する必要があります。クラスセットが変わる場合はsynset.txtを更新してください。実行時ファイルはworkspace/に保存され、RK_CV_WORKSPACEで場所を上書きできます。

ローカルでビルドする

bash

docker build -f docker/rk3576/resnet50v2.dockerfile \
  -t rk3576-resnet50v2:local src/rk3576_resnet50v2

docker build -f docker/rk3588/resnet50v2.dockerfile \
  -t rk3588-resnet50v2:local src/rk3588_resnet50v2

入力と出力

Input: image, video frame, or camera frame. Output: Top-K ImageNet labels and confidence scores.