Как мне перечислить все файлы в каталоге для имени каталога, хранящегося в тензоре?

Question

У меня есть заполнитель, который будет передан строковым значением. Строковое значение представляет путь к некоторому имени каталога. В этом каталоге находятся разные файлы ".jpg". Как я могу попросить TensorFlow перечислить все эти файлы? Другими словами, я хочу добавить узел к моему графу вычислений, который приведет к выводу списка каталогов, если я буду запускать его через мой сеанс. Есть ли способ сделать это?

Answer 1

Вы можете сделать это, используя или не используя TensorFlow. Я только покажу вам, как вы можете перечислить все файлы в определенном каталоге, используя TensorFlow, используя tf.gfile.ListDirectory. Вам даже не нужны заполнители (в данном случае).

import tensorflow as tf


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