Tensorflow实现cnn中的维度问题

Posted 未曾悟道的佛

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Tensorflow实现cnn中的维度问题相关的知识,希望对你有一定的参考价值。

使用TensorFlow实现cnn,其中tf.nn.conv2d(input_tensor…) input_tensor的格式要求是[batch, in_height, in_width, in_channels]
但是python中四维矩阵是[batch, in_channels,in_height, in_width]

使用transpose即可

以上是关于Tensorflow实现cnn中的维度问题的主要内容,如果未能解决你的问题,请参考以下文章

tensorflow框架学习 —— CNN卷积神经网络的实现

具有额外“维度”的图像分类

深度学习之卷积神经网络CNN及tensorflow代码实现示例

《神经网络与深度学习》 卷积神经网络CNN及tensorflow代码实现示例

基于Tensorflow + Opencv 实现CNN自定义图像分类

基于Tensorflow + Opencv 实现CNN自定义图像分类