在Unity中使用Tensorflow的推理图

时间:2018-06-13 07:25:26

标签: c# python unity3d tensorflow machine-learning

首先,让我澄清一些事情,我不知道python。 我正在研究我的最后一年项目,我需要一个很好的物体检测技术,在尝试了很多方法(颜色阈值,Haar-Classifiers)之后,我偶然发现了张量流,发现自己是一个很好的教程,跟着它并得到了我想要的探测器。 问题: 我需要并且我想在Unity上工作,Unity只支持C#。 我找到了一个名为TensorflowSharp的资产,但不知道如何使用它。事实是我不想在Unity上训练,我在python上训练我只需要使用“冻结推理图”(在教程中命名)统一来检测我想要的对象。

请,我必须在一个月内出席,感谢任何帮助。

1 个答案:

答案 0 :(得分:1)

您可以查看我的TFClassify-Unity示例。它使用来自官方Tensorflow回购的训练模型,但您可以通过从' .pb'重新命名其扩展名来尝试自己的模型。 to' .bytes'并用它替换默认值。