modosaic.depth.postprocessor

modosaic.depth.postprocessor

DepthPostprocessor

Bases: ModalityPostprocessor[ndarray]

Postprocessor that saves depth arrays and preview images.

modality property

modality

Return the depth modality.

process

process(record, generated)

Create depth .npy and PNG preview artifacts.

Source code in modosaic/depth/postprocessor.py

@override
def process(self, record: ImageRecord, generated: np.ndarray) -> list[ExperimentArtifact[Any]]:
    """Create depth `.npy` and PNG preview artifacts."""
    logger.debug(f"Post-processing depth for record sample {record.sample_id}")
    depth_image = DepthVisualizationService.depth_to_image(generated)

    return [
        ExperimentArtifact.array(self.sample_path(record, ".npy"), generated),
        ExperimentArtifact.image(self.sample_path(record, ".png"), depth_image),
    ]