Support importing Keras3 as Keras instead of Keras Core

keras_cv/backend/__init__.py

@@ -24,6 +24,8 @@
 
 import types
 
+from packaging.version import parse
+
 from keras_cv.backend.config import multi_backend
 
 # Keys are of the form: "module.where.attr.exists->module.where.to.alias"
@@ -41,7 +43,12 @@
 
 
 if multi_backend():
-    import keras_core as keras
+    import keras
+
+    if not hasattr(keras, "__version__") or parse(keras.__version__) < parse(
+        "3.0"
+    ):
+        import keras_core as keras
 
     keras.backend.name_scope = keras.name_scope
 else:

keras_cv/backend/ops.py

@@ -14,10 +14,16 @@
 from keras_cv.backend.config import multi_backend
 
 if multi_backend():
-    from keras_core.src.backend import vectorized_map  # noqa: F403, F401
-    from keras_core.src.ops import *  # noqa: F403, F401
-    from keras_core.src.utils.image_utils import (  # noqa: F403, F401
-        smart_resize,
-    )
+    try:
+        from keras.src.backend import vectorized_map  # noqa: F403, F401
+        from keras.src.ops import *  # noqa: F403, F401
+        from keras.src.utils.image_utils import smart_resize  # noqa: F403, F401
+    # Import error means Keras isn't installed, or is Keras 2.
+    except ImportError:
+        from keras_core.src.backend import vectorized_map  # noqa: F403, F401
+        from keras_core.src.ops import *  # noqa: F403, F401
+        from keras_core.src.utils.image_utils import (  # noqa: F403, F401
+            smart_resize,
+        )
 else:
     from keras_cv.backend.tf_ops import *  # noqa: F403, F401