nsfw_model_onnx_sample

NSFW判定モデルのGantMan/nsfw_modelのPythonでのONNX推論サンプルです。
変換自体を試したい方は、Google Colaboratory上でConvert2ONNX.ipynbを使用ください。

Requirement

• OpenCV 4.5.3.56 or later
• onnxruntime 1.11.0 or later

Convert

Colaboratoryでノートブックを開き、上から順に実行してください。

Demo

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

python demo_onnx.py

• --image
  画像パスの指定
  デフォルト：sample.jpg
• --model
  ロードするモデルの格納パス
  デフォルト：model/nsfw_mobilenet2_224x224.onnx

Reference

• GantMan/nsfw_model

Author

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

License

nsfw_model_onnx_sample is under MIT License.

Note

サンプルの画像はぱくたそ様の「こちらを見上げる子猫」を使用しています。