tf.keras.preprocessing.image.DirectoryIterator
Итератор, способный считывать изображения из каталога на диске.
Наследуется от: Iterator, PyDataset
tf.keras.preprocessing.image.DirectoryIterator(
directory,
image_data_generator,
target_size=(256, 256),
color_mode='rgb',
classes=None,
class_mode='categorical',
batch_size=32,
shuffle=True,
seed=None,
data_format=None,
save_to_dir=None,
save_prefix='',
save_format='png',
follow_links=False,
subset=None,
interpolation='nearest',
keep_aspect_ratio=False,
dtype=None
)
УСТЕРЕЖДЕН.
| Атрибуты | |
|---|---|
filepaths | Список абсолютных путей к файлам изображений. |
labels | Метки классов каждой наблюдения. |
max_queue_size | |
num_batches | Количество батчей в PyDataset. |
sample_weight | |
use_multiprocessing | |
workers | |
Методы
on_epoch_end
on_epoch_end()
Метод, вызываемый в конце каждой эпохи.
reset
reset()
set_processing_attrs
set_processing_attrs(
image_data_generator,
target_size,
color_mode,
data_format,
save_to_dir,
save_prefix,
save_format,
subset,
interpolation,
keep_aspect_ratio
)
Устанавливает атрибуты для последующей обработки файлов в пакет.
| Аргументы | |
|---|---|
image_data_generator | Экземпляр ImageDataGenerator для использования в случайных преобразованиях и нормализации. |
target_size | Кортеж целых чисел, размеры для изменения размера входных изображений. |
color_mode | Один из "rgb", "rgba", "grayscale". Режим цвета для чтения изображений. |
data_format | Строка, один из channels_first, channels_last. |
save_to_dir | Необязательный каталог для сохранения сгенерированных изображений в визуализируемом формате. Это полезно для визуализации случайных преобразований, применяемых для отладки. |
save_prefix | Префикс строки для сохранения образцов изображений (если save_to_dir задано). |
save_format | Формат для сохранения образцов изображений (если save_to_dir задано). |
subset | Подмножество данных ("training" или "validation"), если validation_split задан в ImageDataGenerator. |
interpolation | Метод интерполяции, используемый для повторной выборки изображения, если размер назначения отличается от размера загруженного изображения. Поддерживаемые методы — "nearest", "bilinear" и "bicubic". Если установлена версия PIL 1.1.3 или более поздняя, также поддерживается "lanczos". Если установлена версия PIL 3.4.0 или более поздняя, также поддерживаются "box" и "hamming". По умолчанию используется "nearest". |
keep_aspect_ratio | Булево значение, необходимо ли изменять размер изображений до целевого размера без искажения соотношения сторон. Изображение обрезается по центру с целевым соотношением сторон перед изменением размера. |
__getitem__
__getitem__(
idx
)
Получает пакет в позиции index.
| Аргументы | |
|---|---|
index | Позиция пакета в PyDataset. |
| Возвращаемое значение | |
|---|---|
| Пакет |
__iter__
__iter__()
__len__
__len__()
| Переменные класса | |
|---|---|
| allowed_class_modes |
{
'binary',
'categorical',
'input',
'sparse',
None
}
|
| white_list_formats | ('png', 'jpg', 'jpeg', 'bmp', 'ppm', 'tif', 'tiff') |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/preprocessing/image/DirectoryIterator