modosaic.normals.postprocessor

modosaic.normals.postprocessor

NormalsPostprocessor

Bases: ModalityPostprocessor[ndarray]

Postprocessor that saves normal fields and RGB previews.

modality property

modality

Return the normals modality.

process

process(record, generated)

Create normal-field .npy and RGB preview artifacts.

Source code in modosaic/normals/postprocessor.py

@override
def process(self, record: ImageRecord, generated: np.ndarray) -> list[ExperimentArtifact[Any]]:
    """Create normal-field `.npy` and RGB preview artifacts."""
    logger.debug(f"Post-processing normals for record sample {record.sample_id}")
    normals_image = NormalsVisualizationService.normals_to_rgb(generated)

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