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

• opencv-python
• onnxruntime

デフォルトでカメラID 0を使用します。

python sample_onnx_inference.py

python sample_onnx_inference.py --video video.mp4

python sample_onnx_inference.py --use_gpu

python sample_onnx_inference.py --output output.mp4

• --video, -v
  入力動画ファイルのパス
  デフォルト：なし（カメラを使用）

• --camera, -c
  カメラデバイス番号の指定
  デフォルト：0

• --output, -o
  出力動画ファイルのパス（省略時は保存なし）
  デフォルト：なし

• --onnx
  ONNXモデルファイルのパス
  デフォルト：onnx_models/retinexformer_lol_v1.onnx

• --use_gpu
  GPU使用フラグ
  デフォルト：False（CPU実行）

• q または ESC キー：終了

onnx_modelsディレクトリに以下のモデルが含まれています：

• retinexformer_lol_v1.onnx（デフォルト）
• retinexformer_lol_v2_real.onnx
• retinexformer_lol_v2_synthetic.onnx
• retinexformer_sid.onnx
• retinexformer_smid.onnx
• retinexformer_sdsd_indoor.onnx
• retinexformer_sdsd_outdoor.onnx
• retinexformer_fivek.onnx
• retinexformer_ntire.onnx

• Retinexformer

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

Retinexformer-ONNX-Sample is under MIT License.