如何在unity mobile中使用tensorflow-gpu

How to use tensorflow-gpu in unity mobile

我正在 Unity 引擎中制作移动 ML 应用程序的原型。
我已经训练了张量流图 (.pb),我想 运行 unity mobile 中的模型。 (android 和 ios)
使用 OpenCVForUnity 插件,使用 dnn 模块,我可以 运行 移动端的 tensorflow 图。但问题是运行宁在CPU。

我需要基于 GPU 的解决方案,OpenCVForUnity 似乎不是合适的方法。
那么对于在 Unity 移动环境中 GPU 上的 运行ning 图有什么想法吗?

您可能想使用 Barracuda, which will allow you to convert a tensorflow model and use it in cross-platform Unity applications. Unity ML-Agents uses Barracuda,这样您就可以使用他们的代码作为如何利用您的神经网络的参考。