ValueError:无法为Tensor'reshape:0'提供形状值(3375,50,50,2),其形状为'(?,5000)'

时间:2017-06-20 09:48:30

标签: tensorflow python-3.5

我正在学习Tensorflow。以下是我使用TensorFlow的MLP代码。我有一些数据维度不匹配的问题。

import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
wholedataset = np.load('C:/Users/pourya/Downloads/WholeTrueData.npz')
data = wholedataset['wholedata'].astype('float32')
label = wholedataset['wholelabel'].astype('float32')
height = wholedataset['wholeheight'].astype('float32')
print(type(data[20,1,1,0]))

learning_rate = 0.001
training_iters = 5
display_step = 20
n_input = 3375

X = tf.placeholder("float32")
Y = tf.placeholder("float32")
weights = {
    'wc1': tf.Variable(tf.random_normal([3, 3, 2, 1])),
    'wd1': tf.Variable(tf.random_normal([3, 3, 1, 1]))
}
biases = {
    'bc1': tf.Variable(tf.random_normal([1])),
    'out': tf.Variable(tf.random_normal([1,50,50,1]))
}
mnist= data

n_nodes_hl1 = 500
n_nodes_hl2 = 500
n_nodes_hl3 = 500

n_classes = 2
batch_size = 100

x = tf.placeholder('float', shape = [None,50,50,2])
shape = x.get_shape().as_list()
dim = np.prod(shape[1:])
x_reshaped = tf.reshape(x, [-1, dim])

y = tf.placeholder('float', shape= [None,50,50,2])
shape = y.get_shape().as_list()
dim = np.prod(shape[1:])
y_reshaped = tf.reshape(y, [-1, dim])

def neural_network_model(data):
    hidden_1_layer = {'weights':tf.Variable(tf.random_normal([5000, 
                       n_nodes_hl1])),
                      'biases':tf.Variable(tf.random_normal([n_nodes_hl1]))}

    hidden_2_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl1, 
                       n_nodes_hl2])),
                      'biases':tf.Variable(tf.random_normal([n_nodes_hl2]))}

    hidden_3_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl2, 
                       n_nodes_hl3])),
                      'biases':tf.Variable(tf.random_normal([n_nodes_hl3]))}

    output_layer = {'weights':tf.Variable(tf.random_normal([n_nodes_hl3, 
                    n_classes])),
                    'biases':tf.Variable(tf.random_normal([n_classes])),}
    l1 = tf.add(tf.matmul(data,hidden_1_layer['weights']), 
        hidden_1_layer['biases'])
    l1 = tf.nn.relu(l1)

    l2 = tf.add(tf.matmul(l1,hidden_2_layer['weights']), 
        hidden_2_layer['biases'])
    l2 = tf.nn.relu(l2)

    l3 = tf.add(tf.matmul(l2,hidden_3_layer['weights']), 
        hidden_3_layer['biases'])
    l3 = tf.nn.relu(l3)

    output = tf.matmul(l3,output_layer['weights']) + output_layer['biases']

    return output
def train_neural_network(x):
    prediction = neural_network_model(x)

    cost = tf.reduce_mean( 
      tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=y) )
    optimizer = tf.train.AdamOptimizer().minimize(cost)

    hm_epochs = 10
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())

        for epoch in range(hm_epochs):
            epoch_loss = 0
            for _ in range(int(n_input/batch_size)):
                epoch_x = wholedataset['wholedata'].astype('float32')
                epoch_y = wholedataset['wholedata'].astype('float32')

                _, c = sess.run([optimizer, cost], feed_dict={x: epoch_x, y: 
                   epoch_y})
                epoch_loss += c

            print('Epoch', epoch, 'completed out 
            of',hm_epochs,'loss:',epoch_loss)

        correct = tf.equal(tf.argmax(prediction, 1), tf.argmax(y, 1))

        accuracy = tf.reduce_mean(tf.cast(correct, 'float'))
        print('Accuracy:',accuracy.eval({x:mnist.test.images, 
        y:mnist.test.labels}))

train_neural_network(x)

我收到以下错误:

ValueError: Cannot feed value of shape (3375, 50, 50, 2) for Tensor 'Reshape:0', which has shape '(?, 5000)'

有谁知道我的代码有什么问题,我该如何解决? 数据值为(3375,50,50,1)

感谢任何人的投入!

1 个答案:

答案 0 :(得分:0)

我认为问题在于您使用相同的变量名称x作为占位符和重新整形,行

x = tf.placeholder('float', shape = [None,50,50,2])

x = tf.reshape(x, [-1, dim])

所以当你

feed_dict={x: your_val}

您正在进行重塑操作的输出。

您应该有不同的名称,例如

x_placeholder = tf.placeholder('float', shape = [None,50,50,2])
x_reshaped = tf.reshape(x, [-1, dim])

然后

feed_dict={x_placeholder: your_val}