From 15b21f029a9e7e375529e819feb8dda85841bf8d Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Fri, 26 Sep 2025 15:40:40 -0700 Subject: [PATCH 1/4] register depth anything presets --- .../depth_anything/depth_anything_presets.py | 36 ++++++++++++++++++- 1 file changed, 35 insertions(+), 1 deletion(-) diff --git a/keras_hub/src/models/depth_anything/depth_anything_presets.py b/keras_hub/src/models/depth_anything/depth_anything_presets.py index 90757d819c..6b33b74cd7 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_presets.py +++ b/keras_hub/src/models/depth_anything/depth_anything_presets.py @@ -1,4 +1,38 @@ """DepthAnything model preset configurations.""" # Metadata for loading pretrained model weights. -backbone_presets = {} +backbone_presets = { + "depth_anything_v2_small": { + "metadata": { + "description": ( + "Small variant of Depth Anything monocular depth estimation " + "(MDE) model trained on synthetic images." + ), + "params": 25_311_169, + "path": "depth_anything", + }, + "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_small/1", + }, + "depth_anything_v2_base": { + "metadata": { + "description": ( + "Base variant of Depth Anything monocular depth estimation " + "(MDE) model trained on synthetic images." + ), + "params": 98_522_945, + "path": "depth_anything", + }, + "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_base/1", + }, + "depth_anything_v2_large": { + "metadata": { + "description": ( + "Large variant of Depth Anything monocular depth estimation " + "(MDE) model trained on synthetic images." + ), + "params": 336_718_529, + "path": "depth_anything", + }, + "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_large/1", + }, +} From ad0b1f02cc8266d3e1f366b28f2139254826f9ae Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Tue, 30 Sep 2025 15:28:14 -0700 Subject: [PATCH 2/4] update description and enable tests --- .../depth_anything_backbone_test.py | 4 +--- .../depth_anything/depth_anything_presets.py | 15 +++++++++------ 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py index 0d0f5e3ce1..4409d4e51a 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py +++ b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py @@ -50,9 +50,8 @@ def test_saved_model(self): ) @pytest.mark.kaggle_key_required - @pytest.mark.extra_large + @pytest.mark.large def test_smallest_preset(self): - self.skipTest("Presets are not uploaded yet.") self.run_preset_test( cls=DepthAnythingBackbone, preset="depth_anything_v2_small", @@ -63,7 +62,6 @@ def test_smallest_preset(self): @pytest.mark.kaggle_key_required @pytest.mark.extra_large def test_all_presets(self): - self.skipTest("Presets are not uploaded yet.") for preset in DepthAnythingBackbone.presets: self.run_preset_test( cls=DepthAnythingBackbone, diff --git a/keras_hub/src/models/depth_anything/depth_anything_presets.py b/keras_hub/src/models/depth_anything/depth_anything_presets.py index 6b33b74cd7..c339b0b4e2 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_presets.py +++ b/keras_hub/src/models/depth_anything/depth_anything_presets.py @@ -5,8 +5,9 @@ "depth_anything_v2_small": { "metadata": { "description": ( - "Small variant of Depth Anything monocular depth estimation " - "(MDE) model trained on synthetic images." + "Small variant of Depth Anything V2 monocular depth estimation " + "(MDE) model trained on synthetic labeled images and real " + "unlabeled images." ), "params": 25_311_169, "path": "depth_anything", @@ -16,8 +17,9 @@ "depth_anything_v2_base": { "metadata": { "description": ( - "Base variant of Depth Anything monocular depth estimation " - "(MDE) model trained on synthetic images." + "Base variant of Depth Anything V2 monocular depth estimation " + "(MDE) model trained on synthetic labeled images and real " + "unlabeled images." ), "params": 98_522_945, "path": "depth_anything", @@ -27,8 +29,9 @@ "depth_anything_v2_large": { "metadata": { "description": ( - "Large variant of Depth Anything monocular depth estimation " - "(MDE) model trained on synthetic images." + "Large variant of Depth Anything V2 monocular depth estimation " + "(MDE) model trained on synthetic labeled images and real " + "unlabeled images." ), "params": 336_718_529, "path": "depth_anything", From e18f17374e6d539a57c4ef9a07652d1a752a1746 Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Tue, 7 Oct 2025 15:45:29 -0700 Subject: [PATCH 3/4] fix expected output shape --- .../depth_anything/depth_anything_backbone_test.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py index 4409d4e51a..4844bcdfe0 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py +++ b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py @@ -52,19 +52,22 @@ def test_saved_model(self): @pytest.mark.kaggle_key_required @pytest.mark.large def test_smallest_preset(self): + image_batch = self.load_test_image(target_size=518)[None, ...] / 255.0 self.run_preset_test( cls=DepthAnythingBackbone, preset="depth_anything_v2_small", - input_data=self.input_data, - expected_output_shape=(2, 70, 70, 1), + input_data=image_batch, + expected_output_shape=(1, 518, 518, 1), ) @pytest.mark.kaggle_key_required @pytest.mark.extra_large def test_all_presets(self): + images = np.ones((2, 518, 518, 3), dtype="float32") for preset in DepthAnythingBackbone.presets: self.run_preset_test( cls=DepthAnythingBackbone, preset=preset, - input_data=self.input_data, + input_data=images, + expected_output_shape=(2, 518, 518, 1), ) From 7ce608469b4fba2aa77057b87e96fc1ec0d7337d Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Wed, 8 Oct 2025 13:50:00 -0700 Subject: [PATCH 4/4] tuple as target img shape --- .../depth_anything/depth_anything_backbone_test.py | 12 ++++++++---- .../depth_anything_depth_estimator_test.py | 5 +++-- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py index 4844bcdfe0..a6b55dfcf0 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py +++ b/keras_hub/src/models/depth_anything/depth_anything_backbone_test.py @@ -52,7 +52,9 @@ def test_saved_model(self): @pytest.mark.kaggle_key_required @pytest.mark.large def test_smallest_preset(self): - image_batch = self.load_test_image(target_size=518)[None, ...] / 255.0 + image_batch = ( + self.load_test_image(target_size=(518, 518))[None, ...] / 255.0 + ) self.run_preset_test( cls=DepthAnythingBackbone, preset="depth_anything_v2_small", @@ -63,11 +65,13 @@ def test_smallest_preset(self): @pytest.mark.kaggle_key_required @pytest.mark.extra_large def test_all_presets(self): - images = np.ones((2, 518, 518, 3), dtype="float32") + image_batch = ( + self.load_test_image(target_size=(518, 518))[None, ...] / 255.0 + ) for preset in DepthAnythingBackbone.presets: self.run_preset_test( cls=DepthAnythingBackbone, preset=preset, - input_data=images, - expected_output_shape=(2, 518, 518, 1), + input_data=image_batch, + expected_output_shape=(1, 518, 518, 1), ) diff --git a/keras_hub/src/models/depth_anything/depth_anything_depth_estimator_test.py b/keras_hub/src/models/depth_anything/depth_anything_depth_estimator_test.py index 1c69756f3f..86a2c0c67a 100644 --- a/keras_hub/src/models/depth_anything/depth_anything_depth_estimator_test.py +++ b/keras_hub/src/models/depth_anything/depth_anything_depth_estimator_test.py @@ -63,8 +63,9 @@ def test_depth_estimator_basics(self): @pytest.mark.large def test_smallest_preset(self): - self.skipTest("Presets are not uploaded yet.") - image_batch = self.load_test_image(target_size=518)[None, ...] / 255.0 + image_batch = ( + self.load_test_image(target_size=(518, 518))[None, ...] / 255.0 + ) self.run_preset_test( cls=DepthAnythingDepthEstimator, preset="depth_anything_v2_small",