/YuNet-ONNX-TFLite-Sample

YuNetのPythonでのONNX、TensorFlow-Lite推論サンプル

Primary LanguagePythonApache License 2.0Apache-2.0

YuNet-ONNX-TFLite-Sample

YuNetのPythonでのONNX、TensorFlow-Lite推論サンプルです。
TensorFlow-LiteモデルはPINTO0309/PINTO_model_zoo/144_YuNetのものを使用しています。

Requirement

  • OpenCV 3.4.2 or later
  • onnxruntime 1.5.2 or later
  • tensorflow 2.6.0 or later

Demo

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

python sample_onnx.py
  • --device
    カメラデバイス番号の指定
    デフォルト:0
  • --movie
    動画ファイルの指定 ※指定時はカメラデバイスより優先
    デフォルト:指定なし
  • --image
    画像ファイルの指定 ※指定時はカメラデバイスや動画より優先
    デフォルト:指定なし
  • --width
    カメラキャプチャ時の横幅
    デフォルト:960
  • --height
    カメラキャプチャ時の縦幅
    デフォルト:540
  • --model
    ロードするモデルの格納パス
    デフォルト:model/face_detection_yunet_120x160.onnx
  • --input_shape
    モデルの入力サイズ
    デフォルト:160,120
  • --score_th
    クラス判別の閾値
    デフォルト:0.6
  • --nms_th
    NMSの閾値
    デフォルト:0.3
  • --topk
    topk指定値
    デフォルト:5000
  • --keep_topk
    keep_topk指定値
    デフォルト:750
python sample_tlite.py
  • --device
    カメラデバイス番号の指定
    デフォルト:0
  • --movie
    動画ファイルの指定 ※指定時はカメラデバイスより優先
    デフォルト:指定なし
  • --image
    画像ファイルの指定 ※指定時はカメラデバイスや動画より優先
    デフォルト:指定なし
  • --width
    カメラキャプチャ時の横幅
    デフォルト:960
  • --height
    カメラキャプチャ時の縦幅
    デフォルト:540
  • --model
    ロードするモデルの格納パス
    デフォルト:model/model_float16_quant.tflite
  • --input_shape
    モデルの入力サイズ
    デフォルト:160,120
  • --score_th
    クラス判別の閾値
    デフォルト:0.6
  • --nms_th
    NMSの閾値
    デフォルト:0.3
  • --topk
    topk指定値
    デフォルト:5000
  • --keep_topk
    keep_topk指定値
    デフォルト:750

Reference

Author

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

License

YuNet-ONNX-TFLite-Sample is under Apache-2.0 License.