cloths_segmentation-onnx-sample

衣服セグメンテーションモデルのternaus/cloths_segmentationのONNX変換/推論のサンプルです。
変換自体を試したい方はColaboratoryなどでConvert2ONNX.ipynbを使用ください。

Requirement

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

Convert ONNX

モデル変換を試す場合は、Colaboratoryでノートブックを開き、上から順に実行してください。

Demo

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

python sample_onnx.py --image=sample.jpg

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --image
  画像ファイルの指定 ※指定時はカメラデバイスや動画より優先
  デフォルト：指定なし
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540
• --model
  ロードするモデルの格納パス
  デフォルト：model/cloths_segmentation.onnx
• --input_resize_rate
  モデル入力時の画像リサイズ割合
  ※処理時間がかかる場合、小さいサイズを指定し、精度を下げて推論速度を向上させる
  デフォルト：1.0

Reference

• ternaus/cloths_segmentation

Author

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

License

cloths_segmentation-onnx-sample is under MIT License.

License(Movie, Image)

サンプルの画像はぱくたそ様の慌てて本社に戻る部長を使用しています。