fix(classifier): duplicated ImageDataGenerator

williamfzc · Apr 1, 2024 · de6bcae · de6bcae
1 parent ac052c9
commit de6bcae
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/stagesepx/classifier/keras.py b/stagesepx/classifier/keras.py
@@ -9,7 +9,12 @@
     import tensorflow
 except ImportError:
     raise ImportError("KerasClassifier requires tensorflow. install it first.")
-from keras.preprocessing.image import ImageDataGenerator
+
+try:
+    from keras.preprocessing.image import ImageDataGenerator
+except ImportError:
+    # https://stackoverflow.com/questions/78145837/importerror-cannot-import-name-imagedatagenerator-from-keras-preprocessing-i
+    from keras.src.legacy.preprocessing.image import ImageDataGenerator
 
 # https://github.com/tensorflow/models/issues/6177
 from tensorflow.keras.models import Sequential