modosaic.normals.generators.base.generator

modosaic.normals.generators.base.generator

NormalsGenerator

NormalsGenerator()

Bases: ModalityGenerator[ndarray], ABC

Base class for surface-normal 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/normals/generators/base/generator.py

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

generate abstractmethod

generate(record)

Generate a normal field for one image record.

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

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