Error topIteration :=> train_img_gen.__next__()

[diouck]

Hello, Very interesting your tutorial. However I have a problem to execute the eration .train_img_gen.next() Do you have an idea to solve this problem? Thank you

[najmehn00]

next() use this

[diouck]

thank you evry thing is ok if you replace classes=None, by classes=['masks'], or classes=['images'],

image_generator = image_datagen.flow_from_directory(
    train_img_path,
    classes=['images'],
    class_mode = None,
    color_mode = 'rgb',
    batch_size = batch_size,
    seed = seed)