我尝试在unity上实现冻结图。但是我有一些错误。我认为这个错误与批量标准化有关,但我不知道如何解决这个问题。
当我训练我的模型时BN是必要的,所以我不能仅仅删除BN。在
因此,我必须找到一些方法,如何在unity上实现时,正确地冻结带有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)
我不太清楚你在问什么或在做什么,但我做了一些研究,希望能给你指明正确的方向。在
Good Audience - TensorFlow + Unity: How to set up a custom TensorFlow graph in Unity
这里有一个人是ML工程师,所以不妨问问他。
Unity Forums - Using Tensorflow's inference graph in Unity
祝你好运。在
相关问题 更多 >
编程相关推荐