python - 如何获取Conv2D层Tensorflow中的步幅值？

Question

我的项目是 CNN 二维模型的可视化，我必须从检查点模型文件中获取 NN 层对象。 Tensorflow是否允许用户获取图层对象？

事实上，我只想要 Conv2d 或 Relu 的步幅或填充值。

Answer 1

仅从检查点获取Layer对象是很困难的。但是您可以通过调用以编程方式获取卷积属性:

tf.get_default_graph().as_graph_def()

...假设您已将检查点恢复到默认图表中。它返回图形节点列表(准确地说是 GraphDef proto)，其中还有 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
    }
  }
}

您只需找到与您感兴趣的特定层相对应的节点即可。