feat(transformers): Add depth estimation pipe

wcrzlh · Jan 21, 2025 · 3b5e316 · 3b5e316
1 parent 8bdd521
commit 3b5e316
Show file tree

Hide file tree

Showing 6 changed files with 602 additions and 2 deletions.
diff --git a/mindone/transformers/__init__.py b/mindone/transformers/__init__.py
@@ -29,6 +29,7 @@
     CLIPVisionModel,
     CLIPVisionModelWithProjection,
 )
+from .models.depth_anything import DepthAnythingForDepthEstimation
 from .models.dpt import DPTForDepthEstimation
 from .models.gemma import (
     GemmaForCausalLM,

diff --git a/mindone/transformers/models/__init__.py b/mindone/transformers/models/__init__.py
@@ -1 +1 @@
-from . import bert, bit, blip_2, clip, dpt, gemma, t5, timesformer, vit, xlm_roberta
+from . import bert, bit, blip_2, clip, depth_anything, dpt, gemma, t5, timesformer, vit, xlm_roberta
diff --git a/mindone/transformers/models/depth_anything/__init__.py b/mindone/transformers/models/depth_anything/__init__.py
@@ -0,0 +1 @@
+from .modeling_depth_anything import DepthAnythingForDepthEstimation
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from . import bert, bit, blip_2, clip, dpt, gemma, t5, timesformer, vit, xlm_roberta
		from . import bert, bit, blip_2, clip, depth_anything, dpt, gemma, t5, timesformer, vit, xlm_roberta
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from .modeling_depth_anything import DepthAnythingForDepthEstimation