Chiangkim

Hello, The issue should be in `ImageDataGenerator` where `classes=[self.image_folder]` should be `classes=None`

问题出在ImageDataGenerator `classes=[self.image_folder]` 应该改成 `classes=None`