kazuhito00/informative-drawings-onnx-sample

Informative DrawingsのPythonでのONNX推論サンプル

Informative-Drawings-ONNX-Sample

Informative DrawingsのPythonでのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。
変換自体を試したい方はColaboratoryなどでInformative-Drawings-Convert2ONNX.ipynbを使用ください。

ONNX Model

以下3種類のモデルをコミットしています

• anime_style_512x512.onnx
• contour_style_512x512.onnx
• opensketch_style_512x512.onnx

Requirement(ONNX推論)

• OpenCV 4.5.3.56 or later
• onnxruntime-gpu 1.9.0 or later
  ※onnxruntimeでも動作しますが、推論時間がかかるのでGPUをお勧めします

Demo

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

python sample_onnx.py

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540
• --model
  ロードするモデルの格納パス
  デフォルト：model/opensketch_style_512x512.onnx
• --input_shape
  モデルの入力サイズ
  デフォルト：512

Reference

• informative-drawings

Author

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

License

Informative-Drawings-ONNX-Sample is under MIT License.

License(Image)

女性の画像はフリー素材ぱくたそ様の写真を利用しています。