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

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

単眼深度推定 Lite-Mono を ONNX に変換して味見👀

Lite-Monoは、軽量 CNN と トランスフォーマー を組み合わせた単眼深度推定だそうです👀

少し動かしてみた感じ、軽さの割に精度が良いと思います。

 

あと、Lite-Mono は公式から以下の8種類のモデルが公開されています。

lite-mono_640x192
lite-mono-small_640x192
lite-mono-tiny_640x192
lite-mono-8m_640x192
lite-mono_1024x320
lite-mono-small_1024x320
lite-mono-tiny_1024x320
lite-mono-8m_1024x320

正直、smallとtinyの違いは見た目では、ほぼ分かりませんでしたが、

色々サイズ違いがあってトレードオフできるのは好みです。

以下は、一番精度高いやつ(lite-mono-8m_1024x320)

 

(2023/04/13追記)

ONNXモデル作り直しました。爆速やった👀

 

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

github.com