Skip to content

register depth anything presets #2420

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Oct 15, 2025
Merged

register depth anything presets #2420

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
15b21f0
register depth anything presets
sachinprasadhs Sep 26, 2025
ad0b1f0
update description and enable tests
sachinprasadhs Sep 30, 2025
e18f173
fix expected output shape
sachinprasadhs Oct 7, 2025
7ce6084
tuple as target img shape
sachinprasadhs Oct 8, 2025
e192d81
add depth_estimation_type
sachinprasadhs Oct 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 35 additions & 1 deletion keras_hub/src/models/depth_anything/depth_anything_presets.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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",
},
}
Loading