如何定义自定义选择的数据集元素进行验证?
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。