我试图使用张量流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)
我希望我的映射函数能够读取给定位置的文件并将它们堆叠成张量。