Функция io.matching_files

Функция io.matching_files предназначена для поиска файлов на диске по заданному шаблону. Она принимает один обязательный параметр - строку с шаблоном (например, 'path/to/files/*.txt'). Функция возвращает тензор строк, содержащий пути к найденным файлам. Если файлы не найдены, возвращается пустой тензор.

Синтаксис

tf.io.matching_files(pattern, [name])

Пример

Давайте создадим несколько временных файлов и найдем их с помощью функции io.matching_files:

import tensorflow as tf
import os
import tempfile

# Create a temporary directory and files
with tempfile.TemporaryDirectory() as tmpdir:
	file1 = os.path.join(tmpdir, 'abcde.txt')
	file2 = os.path.join(tmpdir, 'model.keras')
	
	with open(file1, 'w') as f:
		f.write('test')
	with open(file2, 'w') as f:
		f.write('test')
	
	# Find all files in the directory
	pattern = os.path.join(tmpdir, '*')
	res = tf.io.matching_files(pattern)
	print(res)

Результат выполнения кода:

tf.Tensor([b'/tmp/tmp.../abcde.txt' b'/tmp/tmp.../model.keras'], shape=(2,), dtype=string)

Пример

Давайте найдем только файлы с расширением .txt, используя шаблон с подстановочным знаком:

import tensorflow as tf
import os
import tempfile

with tempfile.TemporaryDirectory() as tmpdir:
	file1 = os.path.join(tmpdir, 'abcde.txt')
	file2 = os.path.join(tmpdir, 'model.keras')
	
	with open(file1, 'w') as f:
		f.write('test')
	with open(file2, 'w') as f:
		f.write('test')
	
	# Find only .txt files
	pattern = os.path.join(tmpdir, '*.txt')
	res = tf.io.matching_files(pattern)
	print(res)

Результат выполнения кода:

tf.Tensor([b'/tmp/tmp.../abcde.txt'], shape=(1,), dtype=string)

Смотрите также

функцию read_file,
которая читает содержимое файла
функцию write_file,
которая записывает данные в файл
функцию get_file,
которая скачивает файл по URL
функцию latest_checkpoint,
которая находит последний чекпоинт