pytomography.mappings.mapping
#
Module Contents#
Classes#
|
- class pytomography.mappings.mapping.MapNet(device='cpu')#
Bases:
torch.nn.Module
MapNet
is the parent class for all mappings used in reconstruction (obj2obj, im2im, obj2im). Subclasses must implement theforward
method.- Parameters:
device (str) – Pytorch device used for computation
- initialize_network(object_meta, image_meta)#
Initalizes the correction network using the object/image metadata
- Parameters:
object_meta (ObjectMeta) – Object metadata.
image_meta (ImageMeta) – Image metadata.
- Return type:
None
- abstract forward(x)#
Abstract method; must be implemented in subclasses to apply a correction to an object/image and return it
- Parameters:
x (torch.tensor) –