0

我正在尝试在统一上实现冻结图。但我有一些错误。我认为与批量标准化有关的错误,但我不知道要解决这个问题。
当我训练我的模型时BN是必要的,所以我不能只删除BN

因此,我必须找到一些方法来正确地用BN冻结图形而不会在统一实现时引起错误。任何的想法?

我已经对这个问题进行了一些研究,我知道不仅BN导致了这个错误,而且 dropout 也有同样的问题。

注意:我正在使用 tensorflowsharp。网络基于 ResNet34,几乎没有修改。

error: TFException: Input 0 of node bn2c_branch2a/Assign was passed float from bn2c_branch2a/mean:0 incompatible with expected float_ref.
TensorFlow.TFStatus.CheckMaybeRaise (TensorFlow.TFStatus incomingStatus, System.Boolean last) (at <013bee3700ac45f2a3b8843fa1d4d914>:0)
TensorFlow.TFGraph.Import (TensorFlow.TFBuffer graphDef, TensorFlow.TFImportGraphDefOptions options, TensorFlow.TFStatus status) (at <013bee3700ac45f2a3b8843fa1d4d914>:0)
TensorFlow.TFGraph.Import (System.Byte[] buffer, TensorFlow.TFImportGraphDefOptions options, TensorFlow.TFStatus status) (at <013bee3700ac45f2a3b8843fa1d4d914>:0)
TensorFlow.TFGraph.Import (System.Byte[] buffer, System.String prefix, TensorFlow.TFStatus status) (at <013bee3700ac45f2a3b8843fa1d4d914>:0)
DeepGes.GClassifier.InitClassifier () (at Assets/Scripts/Gesture_Main/GClassifier.cs:42)
G_Game.Awake () (at Assets/Scripts/Gesture_Main/G_Game.cs:67)
4

1 回答 1

0

我不确定你在问什么或在做什么,但是我为你做了一些研究,希望能指出你正确的方向。

Good Audience - TensorFlow + Unity:如何在 Unity 中设置自定义 TensorFlow 图

这里有一个人是 ML 工程师,所以不妨问问他。
Unity Forums - 在 Unity 中使用 Tensorflow 的推理图

祝你好运。

于 2018-11-04T09:05:38.230 回答