LYT-Net-ONNX-Sample

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

https://github.com/Kazuhito00/LYT-Net-ONNX-Sample/assets/37477845/8ad2245d-51e0-44bc-bd9e-a08551feeac2

Requirement

• OpenCV 4.5.3.56 or later
• onnxruntime 1.13.0 or later

Demo

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

python sample.py --movie=night01.mp4

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

Reference

• albrateanu/LYT-Net

Author

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

License

LYT-Net-ONNX-Sample is under MIT License.

License(Movie)

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