如何列出目录中所有文件的张量中存储的目录名称？

Question

我有一个占位符，它将由字符串值提供。字符串值表示某个目录名称的路径。在该目录中，有不同的“ .jpg”文件。我如何要求TensorFlow列出所有这些文件？ 换句话说，我想在计算图中添加一个节点，如果我在会话中运行该目录，则会列出该目录。有办法吗？

Answer 1

您可以使用TensorFlow来执行此操作。我将仅向您展示如何使用TensorFlow和tf.gfile.ListDirectory列出特定目录中的所有文件。您甚至不需要占位符（在这种情况下）。

import tensorflow as tf


dir_path = "."
print(tf.gfile.ListDirectory(dir_path))