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

Question

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

标签： filetensorflow

Answer 1

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

import tensorflow as tf


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