Wie erhält man das Gewichtungsformat vom TensorFlow .pb-Modell?

8

Ich möchte die Knoten des Tensorflow-.pb-Modells reorganisieren, also erhalte ich NodeDef von GraphDef, und benutze NodeDef.attr () für den Knoten von "Conv2D".    Ich kann Parameter wie strides, padding, data_format, use_cudnn_on_gpu von attr erhalten, aber kann nicht die Gewichte Formatparameter erhalten.    Die Sprache, die ich verwende, ist C ++.    Wie man es bekommt! Danke!

    
yiyijing 12.07.2017, 07:52
quelle

1 Antwort

4

Conv2D hat zwei Eingänge: der erste ist Daten und der zweite ist filter (oder Gewichte), so dass Sie einfach das Format der zweiten Eingabe von Conv2D überprüfen können. Wenn Sie C ++ verwenden, können Sie Folgendes versuchen:

%Vor%     
Max 19.10.2017 05:28
quelle

Tags und Links