unity3d - 如何在统一移动设备中使用 tensorflow-gpu

Question

我正在 Unity 引擎中对移动机器学习应用程序进行原型设计。
我已经训练了张量流图（.pb），我想在统一移动设备中运行模型。(android 和 ios)
使用OpenCVForUnity插件和 dnn 模块，我可以在移动设备上运行 tensorflow 图。但问题是它在 CPU 上运行。

我需要基于 GPU 的解决方案，而且 OpenCVForUnity 似乎不是解决这个问题的合适方法。
那么在统一移动环境中在 GPU 上运行图形有什么想法吗？

Answer 1

您可能想要使用Barracuda，它允许您转换 tensorflow 模型并将其用于跨平台的 Unity 应用程序。Unity ML-Agents使用梭子鱼，因此您可以使用他们的代码作为如何利用您的神经网络的参考。