我在用
file_data = tf.data.Dataset.list_files("../*.png")
收集图像文件以在 TensorFlow 中进行训练，但希望访问收集的文件名列表，以便执行标签查找。
调用 sess.run([file_data]) 一直不成功：
TypeError: Fetch argument <TensorSliceDataset shapes: (), types: tf.string> has invalid type <class 'tensorflow.python.data.ops.dataset_ops.TensorSliceDataset'>, must be a string or Tensor. (Can not convert a TensorSliceDataset into a Tensor or Operation.)
我还可以使用其他方法吗？
通过一些额外的实验，我找到了解决这个问题的方法：
首先，将 Dataset 转换为迭代器：
iterator_helper = file_data.make_one_shot_iterator()
然后，迭代 tf Session 中的元素：
with tf.Session() as sess:
filename_temp = iterator_helper.get_next()
print(sess.run[filename_temp])
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
