Question

我正在尝试统一实现冻结图。但是我有一些错误。我认为该错误与批处理规范化有关，但我不知道要解决此问题。
训练模型时，必须使用 BN ，所以我不能只删除 BN 。

因此，我必须找到一种方法来正确地使用 BN 冻结图，而在统一实现时不会引起错误。有想法吗？

我已经对该问题进行了一些研究，我不仅知道 BN 会导致此错误，而且辍学也存在同样的问题。

注意：我使用的是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)

Answer 1

我不确定您要问或正在做什么，但是我为您做过一些研究，希望可以为您指明正确的方向。

Good Audience - TensorFlow + Unity: How to set up a custom TensorFlow graph in Unity

这里有一个人是ML工程师，所以也许可以问问他。
Unity Forums - Using Tensorflow's inference graph in Unity

好运。

张量流批量归一化冻结

1 个答案: