Resolve comments.

Author: james77777778

keras_hub/src/models/depth_anything/depth_anything_backbone.py

@@ -58,7 +58,7 @@ class DepthAnythingBackbone(Backbone):
     input_data = {
         "images": np.ones(shape=(1, 518, 518, 3), dtype="float32"),
     }
-    model = keras_hub.models.DINOV2Backbone.from_preset(
+    model = keras_hub.models.DepthAnythingBackbone.from_preset(
         "depth_anything_v2_small"
     )
     model(input_data)
@@ -67,7 +67,7 @@ class DepthAnythingBackbone(Backbone):
     input_data = {
         "images": np.ones(shape=(1, 224, 224, 3), dtype="float32"),
     }
-    model = keras_hub.models.DINOV2Backbone.from_preset(
+    model = keras_hub.models.DepthAnythingBackbone.from_preset(
         "depth_anything_v2_small", image_shape=(224, 224, 3)
     )
     model(input_data)

keras_hub/src/models/dinov2/dinov2_layers.py

@@ -277,16 +277,16 @@ def get_config(self):
         )
         return config
 
-    def compute_output_shape(self, inputs_shape):
-        output_shape = [inputs_shape[0], None, self.hidden_dim]
+    def compute_output_shape(self, input_shape):
+        output_shape = [input_shape[0], None, self.hidden_dim]
         if self.data_format == "channels_last":
-            if inputs_shape[1] is not None and inputs_shape[2] is not None:
-                patch_num = inputs_shape[1] // self.patch_size
+            if input_shape[1] is not None and input_shape[2] is not None:
+                patch_num = input_shape[1] // self.patch_size
                 # 1 is for cls token.
                 output_shape[1] = 1 + self.num_register_tokens + patch_num**2
         else:
-            if inputs_shape[2] is not None and inputs_shape[3] is not None:
-                patch_num = inputs_shape[2] // self.patch_size
+            if input_shape[2] is not None and input_shape[3] is not None:
+                patch_num = input_shape[2] // self.patch_size
                 # 1 is for cls token.
                 output_shape[1] = 1 + self.num_register_tokens + patch_num**2
         return output_shape