Skip to content

modosaic.depth.generators.impl.depth_anything_v2_metric_small

modosaic.depth.generators.impl.depth_anything_v2_metric_small

DepthAnythingV2MetricSmall 

DepthAnythingV2MetricSmall()

Bases: DepthGenerator

Metric Depth Anything V2 Small generator.

Load the pinned Hugging Face metric-depth pipeline.

Source code in modosaic/depth/generators/impl/depth_anything_v2_metric_small.py 
20
21
22
23
24
25
26
27
28
def __init__(self):
    """Load the pinned Hugging Face metric-depth pipeline."""
    super().__init__()
    self.model_specs = HFModelSpec(
        model_id="depth-anything/Depth-Anything-V2-Metric-Outdoor-Small-hf",
        revision="fd2c22027eaf20374204f14099b8341e1925ad39",
        dtype=self.pipeline_dtype,
    )
    self.pipeline = self._load_pipeline()

generate 

generate(record)

Generate a metric depth map for one image record.

Source code in modosaic/depth/generators/impl/depth_anything_v2_metric_small.py 
40
41
42
43
44
45
46
47
48
@override
def generate(self, record: ImageRecord) -> np.ndarray:
    """Generate a metric depth map for one image record."""
    logger.debug(f"Generating depth anything v2 metric small image for record sample {record.sample_id}")
    img = ImageService.bytes_to_pil(record.image_bytes)
    depth_tensor = self.pipeline(img)["predicted_depth"]

    np_depth = depth_tensor.detach().cpu().float().numpy()
    return np_depth