私のプロジェクトはCNN2dモデルの視覚化であり、チェックポイントモデルファイルからNNレイヤーオブジェクトを取得する必要があります。Tensorflowを使用すると、ユーザーはレイヤーオブジェクトを取得できますか?
実際、Conv2dまたはReluからのストライドまたはパディング値が必要です。
Layer
チェックポイントだけからオブジェクトを取得するのは困難です。ただし、次を呼び出すことにより、プログラムで畳み込み属性を取得できます。
tf.get_default_graph().as_graph_def()
...チェックポイントをデフォルトのグラフに復元したと仮定します。グラフノードのリスト(GraphDef
正確にはプロト)を返します。その中にはConv2D
、次のようなノードもあります。
node {
name: "conv2d/Conv2D"
op: "Conv2D"
input: "input_layer"
input: "conv2d/kernel/read"
attr {
key: "T"
value {
type: DT_FLOAT
}
}
attr {
key: "data_format"
value {
s: "NHWC"
}
}
attr {
key: "padding"
value {
s: "SAME"
}
}
attr {
key: "strides"
value {
list {
i: 1
i: 1
i: 1
i: 1
}
}
}
attr {
key: "use_cudnn_on_gpu"
value {
b: true
}
}
}
興味のある特定のレイヤーに対応するノードを見つける必要があります。
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加