如何在Unity中使用TensorFlow Lite模型

问题描述 投票:2回答:1

[我想在Unity中编写一个Android游戏,该游戏利用了我已经存在的,经过预先训练的神经网络,而我仅将其作为.tflite文件。

我的第一个计划涉及将.tflite文件转换为.nn文件,然后可以由Unity的ml-agents使用。但是,到目前为止,我只找到了一种方法来将冻结的图(.pb)转换为.nn,并且找不到从.tflite转换为.pb的好方法。解决该问题的其他想法也将不胜感激。

android unity3d tensorflow-lite ml-agent
1个回答
0
投票

我使用Unity制作了一个TensorFlow Lite示例。我主要在iOS和macOS上进行了测试,但应该可以在Android上运行。

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

© www.soinside.com 2019 - 2024. All rights reserved.