modosaic.depth.generators.base.generator

modosaic.depth.generators.base.generator

DepthGenerator

DepthGenerator()

Bases: ModalityGenerator[ndarray], ABC

Base class for depth-map generators.

Attributes:

Name	Type	Description
device	device	Torch device selected for inference.
dtype	dtype	Torch dtype selected for inference.

Initialize device and dtype defaults.

Source code in modosaic/depth/generators/base/generator.py

def __init__(self) -> None:
    """Initialize device and dtype defaults."""
    self.device, self.dtype = DeviceService.get_device_type()

pipeline_dtype property

pipeline_dtype

Return a dtype compatible with Transformers depth pipeline postprocessing.

generate abstractmethod

generate(record)

Generate a depth map for one image record.

Source code in modosaic/depth/generators/base/generator.py

@override
@abstractmethod
def generate(self, record: ImageRecord) -> np.ndarray:
    """Generate a depth map for one image record."""
    raise NotImplementedError()