Add dict check for deserialization (#2286)

keras-team · Jan 8, 2024 · dd4bef2 · dd4bef2
1 parent 7738e2f
commit dd4bef2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py b/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py
@@ -655,12 +655,14 @@ def from_config(cls, config):
             config["backbone"]
         )
         label_encoder = config.get("label_encoder")
-        if label_encoder is not None:
+        if label_encoder is not None and isinstance(label_encoder, dict):
             config["label_encoder"] = keras.saving.deserialize_keras_object(
                 label_encoder
             )
         prediction_decoder = config.get("prediction_decoder")
-        if prediction_decoder is not None:
+        if prediction_decoder is not None and isinstance(
+            prediction_decoder, dict
+        ):
             config[
                 "prediction_decoder"
             ] = keras.saving.deserialize_keras_object(prediction_decoder)