https://github.com/kazuhito00/openlenda-onnx-inference-sample
OpenLendaのPythonでのONNX推論サンプル
https://github.com/kazuhito00/openlenda-onnx-inference-sample
onnx onnxruntime opencv openlenda python traffic-light traffic-light-detection
Last synced: 3 months ago
JSON representation
OpenLendaのPythonでのONNX推論サンプル
- Host: GitHub
- URL: https://github.com/kazuhito00/openlenda-onnx-inference-sample
- Owner: Kazuhito00
- License: apache-2.0
- Created: 2023-10-28T07:23:25.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2023-10-29T01:08:46.000Z (over 1 year ago)
- Last Synced: 2025-04-23T00:44:26.990Z (3 months ago)
- Topics: onnx, onnxruntime, opencv, openlenda, python, traffic-light, traffic-light-detection
- Language: Python
- Homepage:
- Size: 52.7 MB
- Stars: 5
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# openlenda-onnx-inference-sample
[OpenLenda（日本の信号機検出＋認識）](https://github.com/turingmotors/openlenda)のPythonでのONNX推論サンプルです。
# Requirement
* OpenCV 4.8.1.78 or later
* onnxruntime 1.14.1 or later ※GPU推論する際は「onnxruntime-gpu」
# Demo
デモの実行方法は以下です。
```bash
python sample_onnx.py --image=sample.jpg
```
* --device
カメラデバイス番号の指定
デフォルト：0
* --movie
動画ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト：指定なし
* --image
画像ファイルの指定 ※指定時はカメラデバイス・動画ファイルより優先
デフォルト：指定なし
* --model
ロードするモデルの格納パス
デフォルト：model/openlenda_s.onnx
* --score_th
スコア閾値
デフォルト：0.5
* --nms_th
NMS閾値
デフォルト：0.01
* --grouping_th
同じバウンディングボックスの結果をグルーピングするIOU閾値
デフォルト：0.8
* --use_gpu
GPU推論
デフォルト：指定なし
# ToDo
* [ ] マルチクラスNMSの暫定実装の見直し
# Reference
* [turingmotors/openlenda](https://github.com/turingmotors/openlenda)
# Author
高橋かずひと(https://twitter.com/KzhtTkhs)
# License
openlenda-onnx-inference-sample is under [Apache2.0 License](LICENSE).
# License(Image)
サンプル画像は[ぱくたそ](https://www.pakutaso.com/)様の[信号と車のライトがカラフルに反射する雨の道路](https://www.pakutaso.com/20230304087post-46153.html)を使用しています。