Keras自定义验证生成器

时间:2020-03-12 07:09:21

标签: keras

如何定义自定义选择的数据集元素进行验证?

train_datagen = ImageDataGenerator(
        rotation_range=90,
        # ...
        validation_split=0.1)

train_generator = train_datagen.flow_from_directory(
        PATH + 'train',
        # ...
        subset='training')

validation_generator = train_datagen.flow_from_directory(
        PATH + 'train',
        # ...
        subset='validation')

我需要添加具有唯一前缀名称的文件。从名称为image = 00_00 *的文件中,我应该只取一个:

image_00_00_01.jpg
image_00_00_02.jpg
image_00_00_03.jpg
...

例如image_00_00_03.jpg。

0 个答案:

没有答案