Как для-l oop tf.strings?

Question

Я создал список путей к моему набору данных:

list_train = tf.data.Dataset.list_files(str(TRAIN_DIR + '/*/*'))

Я хочу создать этот список для -l oop, чтобы преобразовать его в пару (изображение, метка). Я хочу использовать функцию с веб-сайта TensorFlow:

def get_labels(file_path):
    #convert the path to a list of path components
    parts = tf.strings.split(file_path, os.path.sep)
    #my class directory is the second to last
    return parts[-3] == CLASS_NAMES

Но эта функция не принимает набор данных 'list_train'.

Answer 1

Я обнаружил, что не должен ничего добавлять к этой функции. Эта функция передается в другую:

def process_path(file_path):
  label = get_label(file_path)
  return label

И затем мы используем ее следующим образом:

labeled_dataset = list_train.map(process_path, num_parallel_calls=AUTOTUNE) 

Весь процесс очень хорошо объяснен здесь: