pytomography.mappings.mapping#

Module Contents#

Classes#

MapNet

MapNet is the parent class for all mappings used in reconstruction (obj2obj, im2im, obj2im). Subclasses must implement the forward method.

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 the forward 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:
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) –