Skip to content

背景除去モデルであるDIS/IS-NetのPythonでのONNX推論サンプル

License

Notifications You must be signed in to change notification settings

Kazuhito00/DIS-ONNX-Sample

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DIS-ONNX-Sample

背景除去モデルであるDIS/IS-NetのPythonでのONNX推論サンプルです。
変換自体を試したい方は、Google Colaboratory上でConvert2ONNX.ipynbを使用ください。

image

Requirement

  • OpenCV 4.5.3.56 or later
  • onnxruntime 1.8.0 or later
  • gdown 4.6.4 or later ※「model/isnet-general-use.onnx」ファイルが無い場合

Demo

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

python demo_DIS_onnx.py
  • --device
    カメラデバイス番号の指定
    デフォルト:0
  • --movie
    動画ファイルの指定 ※指定時はカメラデバイスより優先
    デフォルト:指定なし
  • --model
    ロードするモデルの格納パス
    デフォルト:model/isnet-general-use.onnx
  • --score_th
    マスク値の閾値 ※指定する場合は0.5など小数値を指定
    デフォルト:None

Reference

Author

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

License

DIS-ONNX-Sample is under Apache 2.0 License.