нормализация заморозки - PullRequest
Новая
       15

нормализация заморозки

0 голосов
/ 04 ноября 2018

Я пытаюсь реализовать замороженный граф на единице. Но у меня есть ошибка. Я думаю, что ошибка связана с нормализацией партии, но я понятия не имею, чтобы решить эту проблему.
Когда я тренируюсь, моя модель BN необходима, поэтому я не могу просто удалить BN .

Так что мне нужно найти способ как правильно заморозить график с помощью BN , не вызывая ошибки при его реализации на единице. Любая идея?

У меня уже было несколько исследований для этой проблемы, я знаю, что не только BN причина этой ошибки, но и выпадение имеет ту же проблему.

примечание: я использую тензор потока резкости. сеть основана на 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)

1 Ответ

0 голосов
/ 04 ноября 2018

Я не совсем уверен в том, что вы спрашиваете или над чем работаете, однако я провел некоторое исследование для вас, чтобы, надеюсь, направить вас в правильном направлении.

Хорошая аудитория - TensorFlow + Unity: как настроить собственный график TensorFlow в Unity

Здесь есть парень, который является инженером ML, так что, возможно, попробуйте спросить его.
Форумы Unity - Использование графа вывода Tensorflow в Unity

Удачи.

...