Merge: [Efficientnet/TF2] fix keras imports

Author: nv-kkudrynski

TensorFlow2/Classification/ConvNets/dataloader/augment.py

@@ -26,7 +26,11 @@
 import tensorflow as tf
 from typing import Any, Dict, List, Optional, Text, Tuple
 
-from keras.layers.preprocessing import image_preprocessing as image_ops
+try:
+    from keras.layers.preprocessing import image_preprocessing as image_ops
+except (ImportError, ModuleNotFoundError):
+    import keras.src.layers.preprocessing.image_preprocessing as image_ops
+
 
 # This signifies the max integer that the controller RNN could predict for the
 # augmentation scheme.