lyuwenyu/RT-DETR(v2)のPythonでのONNX推論サンプルです。
変換自体を試したい方は、Google Colaboratory上でConvert2ONNX.ipynbを使用ください。
- OpenCV 4.5.3.56 or later
- onnxruntime 1.11.0 or later
Colaboratoryでノートブックを開き、上から順に実行してください。
デモの実行方法は以下です。
python sample_onnx.py- --device
カメラデバイス番号の指定
デフォルト：0 - --movie
動画ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト：指定なし - --image
画像ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト：指定なし - --model
ロードするモデルの格納パス
デフォルト：model/rtdetrv2_r18vd_120e_coco_rerun_48.onnx - --score_th
検出閾値
デフォルト：0.6 - --unuse_gpu
GPU推論なし（CPU推論）
デフォルト：指定なし
高橋かずひと(https://twitter.com/KzhtTkhs)
RT-DETR-ONNX-Sample is under Apache 2.0 License.
サンプルの画像はぱくたそ様の「ものすごい数のバイクの列（ベトナムホーチミン）」を使用しています。