a question regarding customdatagenerator

[wenouyang]

Hi Raghavendra,

Thanks for sharing the code. I have a question regarding generator.py. Looks like CustomDataGenerator(Iterator): only contains two methods: init and next. You mentioned that this is modified Modified keras ImageDataGenerator, which seems to me includes a lot of other methods.

Or in specific, I am just trying to understand how does your CustomDataGenerator(Iterator) work?. Thanks.

Besides, in train.py, the train_generator and val_generator are setup in different manner, may I know the difference?

train_generator = CustomDataGenerator(X_train, y_train, transform, batch_size)

val_generator = CustomDataGenerator(X_val, y_val, lambda x, y: transform(x, y, augment=False), batch_size)