BiRefNetをONNXに変換して味見👻 - 高橋かずひとのプログラミング、その他、備忘録。

BiRefNetをONNXに変換して動作を味見しています。
いわゆるDIS（Dichotomous Image Segmentation）モデルです。
背景除去、背景分離、Salient Object Detectionとか呼ばれたりもする？
Salient Object Detectionは違うか🤔？

ONNX変換後の重みファイルが1GB弱あって、それなりに重いです。
重いなりの検出性能は感じますが👀

BiRefNetをONNXに変換して味見している👀
いわゆるDIS（Dichotomous Image Segmentation）モデルです。
重いですが、重いだけの性能はある感じ🦔 pic.twitter.com/TtL7Q7Xjhq

— 高橋 かずひと@闇のパワポLT職人 (@KzhtTkhs) 2024年8月14日

ONNX変換にあたり、deform_conv2dが邪魔だったので、以下のリポジトリに助けてもらいました。

github.com

ソースコードは以下にコミットしています。

github.com