Kazuhito00 / LDC-ONNX-Sample

LDC: Lightweight Dense CNN for Edge DetectionのPythonでのONNX推論サンプル

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LDC-ONNX-Sample

LDC: Lightweight Dense CNN for Edge DetectionのPythonでのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。
変換自体を試したい方はColaboratoryでLDC-Convert2ONNX.ipynbを使用ください。


1枚目:元画像 2枚目:Fuse画像(LDC_3840x2160.onnx使用) 3枚目:Average画像(LDC_3840x2160.onnx使用)

Requirement

  • OpenCV 4.5.3.56 or later
  • onnxruntime 1.13.0 or later

Demo

デモの実行方法は以下です。

python demo_onnx.py
  • --device
    カメラデバイス番号の指定
    デフォルト:0
  • --movie
    動画ファイルの指定 ※指定時はカメラデバイスより優先
    デフォルト:指定なし
  • --image
    画像ファイルの指定 ※指定時はカメラデバイス・動画ファイルより優先
    デフォルト:指定なし
  • --model
    ロードするモデルの格納パス
    デフォルト:model/LDC_640x360.onnx

Reference

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

LDC-ONNX-Sample is under CC-BY-NC 4.0.

License(Movie)

サンプル動画はNHKクリエイティブ・ライブラリーイギリス オックスフォードのクライスト・チャーチ学生寮外観を使用しています。

About

LDC: Lightweight Dense CNN for Edge DetectionのPythonでのONNX推論サンプル

License:Other


Languages

Language:Python 55.9%Language:Jupyter Notebook 44.1%