Is it possible to get the file names that were loaded using flow_from_directory
?
I have :
datagen = ImageDataGenerator(
rotation_range=3,
# featurewise_std_normalization=True,
fill_mode='nearest',
width_shift_range=0.2,
height_shift_range=0.2,
horizontal_flip=True
)
train_generator = datagen.flow_from_directory(
path+'/train',
target_size=(224, 224),
batch_size=batch_size,)
I have a custom generator for my multi output model like:
a = np.arange(8).reshape(2, 4)
# print(a)
print(train_generator.filenames)
def generate():
while 1:
x,y = train_generator.next()
yield [x] ,[a,y]
Node that at the moment I am generating random numbers for a
but for real training , I wish to load up a json
file that contains the bounding box coordinates for my images. For that I will need to get the file names that were generated using train_generator.next()
method. After I have that , I can load the file, parse the json
and pass it instead of a
. It is also necessary that the ordering of the x
variable and the list of the file names that I get is the same.
See Question&Answers more detail:
os 与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…