Tensorflow加载numpy数组 - 形状错误

时间:2017-02-08 13:14:19

标签: numpy tensorflow

因此,当我保存一个int16 numpy数组并重新加载它时,我获得了与预期相同的形状。在这种特殊情况下,形状为(335,306,306),产生的尺寸为31368060.

现在我正在尝试将这样的数组加载到tensorflow管道中,如下所示;

def loadFiles():
    imgPaths = glob.glob("*.npy") # List containing the one numpy array

    Q = tf.train.string_input_producer(imgPaths,num_epochs=10,shuffle=True)
    reader = tf.WholeFileReader()
    key, value = reader.read(Q)
    image_bytes = tf.decode_raw(value,tf.int16)
    return key, image_bytes

在我得到的会话中评估相同的示例;

p,img = loadFiles()
 with tf.Session() as sess:
        tf.global_variables_initializer().run()
        tf.local_variables_initializer().run()
        coord = tf.train.Coordinator()
        threads = tf.train.start_queue_runners(coord=coord,sess=sess)
        im = img.eval()
        print(im.shape)

Yielding(31368100,)与(31368060,)不同。出了什么问题?

0 个答案:

没有答案