memomem

備忘録・メモ置き場

Tensorflow -> ONNX

Teachable MachineのデータをUnity Barracudaを使うためにTensforflowからONNXに変換をおこなう。

Macでtensorflow-onnxを実行していたが、エラーでコンバードできず。

結局Anacondaで仮想環境を作って、Pythonのバージョンを変更

  • Python 3.7.16
  • tensorflow 2.11.0
  • onnx 1.14.1
  • tf2onnx 1.15.1

で変換できるようになった。

pip install tensorflow
pip install onnxruntime
pip install -U tf2onnx
python -m tf2onnx.convert --saved-model [tensorflow-model-path] --output [model.onnx]

参考

https://github.com/onnx/tensorflow-onnx

techceed-inc.com