PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
https://qiita.com/PINTO
MIT License
3.55k stars 568 forks source link

Add DeDoDe #357

Closed fabio-sim closed 1 year ago

fabio-sim commented 1 year ago

こんにちは、@PINTO0309 さん。お世話になっております。

この前、LightGlue-ONNXをツイートしてくれて、誠にありがとうございました。今回DeDoDeと言うkeypointモデルをONNX化しました。もし良かったら、是非model zooに追加して下さい!

PINTO0309 commented 1 year ago

ありがとうございます! とてもいいですね。今着手してる眼球6DoFモデルの作業が終わったらモデル本体も引用させていただいたうえで改造コミットしてみます。TensorRTのモデルは入力解像度や出力キーポイント数が非固定だとオプティマイゼーションが毎フレームで発生してしまう問題や、オプティマイゼーションの回数がおおむね200回を超える(200フレームぐらい推論したとき)に TensorRT Engine がAbortしてしまう問題がありますので、ちょっとだけ調整させていただく可能性があります。