DIS(Dichotomous Image Segmentation)モデルであるZhengPeng7/BiRefNetのPythonでのONNX推論サンプルです。
変換自体を試したい方は、Google Colaboratory上でConvert2ONNX.ipynbを使用ください。
Note
deform_conv2dをONNXへ変換するためにmasamitsu-murase/deform_conv2d_onnx_exporterを利用しています。
deform_conv2dはONNXでサポートされていないオペレーターのため、別のオペレーターで代替しています
その結果、オリジナルモデルと比べ、推論速度や精度が劣る可能性があります
Colaboratoryでノートブックを開き、上から順に実行してください。
※ハイメモリモード必須
デモの実行方法は以下です。
python demo_onnx.py
高橋かずひと(https://twitter.com/KzhtTkhs)
LicenseBiRefNet-ONNX-Sample is under MIT License.
Noteサンプルの画像はぱくたそ様の「攻撃ヘリコプターアパッチ(AH-64)」を使用しています。
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4