modosaic.image.postprocessor

modosaic.image.postprocessor

ImagePostprocessor

Bases: ModalityPostprocessor[bytes]

Postprocessor that saves source images as PNG artifacts.

modality property

modality

Return the image modality.

process

process(record, generated)

Convert encoded image bytes into a PNG artifact.

Source code in modosaic/image/postprocessor.py

@override
def process(self, record: ImageRecord, generated: bytes) -> list[ExperimentArtifact[Any]]:
    """Convert encoded image bytes into a PNG artifact."""
    logger.debug(f"Post-processing source image for record sample {record.sample_id}")
    image = ImageService.bytes_to_pil(generated)
    return [
        ExperimentArtifact.image(self.sample_path(record, ".png"), image)
    ]