Open4

LiteRT を Unityで動かせるか検討

Koki IbukuroKoki Ibukuro

https://ai.google.dev/edge/litert

TensorFlow LiteがLiteRTという名前になった。リポジトリもTensorFlowのサブディレクトリから個別のリポジトリへと別れて開発が進んでいる。

現状の公開版のAPIはTensorFlow Liteのそのままだが、Next とよばれるAPIはシンプルに使いやすく見える。
https://ai.google.dev/edge/litert/next/overview

TensorFlow LiteをUnityで動かすことを5年ほどやっているが、LiteRTへ移行するか検討してみてもいいかもしれない。

https://github.com/asus4/tf-lite-unity-sample

Koki IbukuroKoki Ibukuro

TensorFlow Liteで画像が絡む処理をやろうとすると、モデル本体の処理よりも、GPU <-> CPU間のデータ転送コストのほうがボトルネックになりがちだったので、このようにシンプルにGPU Bufferを直接割当できるなら、それだけで移行の価値はあるかもしれない…。

https://github.com/google-ai-edge/LiteRT/blob/bfb1db748d4d8e2dbb71da025b90fc2c2e78e1c8/litert/samples/async_segmentation/segmentation_model.cc#L147-L162