使用张量字符串输入的Tensorflow文件操作

时间:2018-01-06 12:11:02

标签: input tensorflow dataset python-multithreading tensor

我试图使用张量流tf.gfile函数,例如ListDirectory,walk等。 我想知道为什么他们不能有张量字符串输入?我错过了什么吗?

我正在尝试从文件夹中读取所有图像并将张量堆叠为输入数据。

def input_parser(dir_name, label):
    ret_stack = []
    for parent, dirnames, filenames in tf.gfile.Walk(dir_name):
    ...
       img = file_read_operation(filenames)
       ret_stack.append(img)
    return tf.stack(ret_stack), label
...

#train_dir list including folder names
tr_data = Dataset.from_tensor_slices((train_dir, train_labels))
tr_data = tr_data.map(input_parser)
iterator = Iterator.from_structure(tr_data.output_types, tr_data.output_shapes)
next_element = iterator.get_next()

training_init_op = iterator.make_initializer(tr_data)
...
...
elem1, elem2 = sess.run(next_element)

我希望我的映射函数能够读取给定位置的文件并将它们堆叠成张量。

0 个答案:

没有答案