输入尺寸为cnn

时间:2018-03-11 03:47:50

标签: image classification conv-neural-network

在图像中从4d输入(训练图像的数量 x y * z)向CNN提供3dimension(x y z)输入图像有什么区别分类

2 个答案:

答案 0 :(得分:0)

通常,CNN的输入是[批次,标签,图像,频道]。您可以使用numpy或tensor来重塑输入。

答案 1 :(得分:0)

TensorFlow的tf.nn.conv2d op采用4D输入张量[batch, in_height, in_width, in_channels]。这里的第一个参数是批处理中的图像数。其他参数是定义每个图像的参数。

https://www.tensorflow.org/api_docs/python/tf/nn/conv2d