pytomography.mappings.mapping
#
Module Contents#
Classes#
|
- class pytomography.mappings.mapping.MapNet(device='cpu')[source]#
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)[source]#
Initalizes the correction network using the object/image metadata
- Parameters:
object_meta (ObjectMeta) – Object metadata.
image_meta (ImageMeta) – Image metadata.
- Return type:
None