高橋かずひとのプログラミング、その他、備忘録。

日々調べてたことや、作ってみたものをメモしているブログ。 お決まりの断り文句ですが、このブログに書かれている内容は個人の見解であり、所属する組織の公式見解ではありません。チラ裏。

BiRefNetをONNXに変換して味見👻

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

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

 

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

github.com


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

github.com