SCI-ONNX-Sample

Low-Light Image EnhancementモデルであるSCIのPythonでのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。変換自体を試したい方はConvert2ONNX.ipynbを使用ください。

SCI-Demo.mp4

Requirement

• OpenCV 4.5.3.56 or later
• onnxruntime 1.8.0 or later
• Pytorch 1.8 or later ※ONNX変換を実施する場合のみ

Demo

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

python sample.py

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --model
  ロードするモデルの格納パス
  デフォルト：model/sci_difficult_512x512.onnx
• --input_shape
  モデルの入力サイズ
  デフォルト：512,512

Reference

• tengyu1998/SCI

Author

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

License

SCI-ONNX-Sample is under MIT License.

License(Movie)

サンプル動画はNHKクリエイティブ・ライブラリーの雨イメージ　夜の道路を走る車を使用しています。