HomogeneousTransformationLayer

class HomogeneousTransformationLayer(n_dims: int, use_cpp=False)

Bases: sphinx.ext.autodoc.importer._MockObject

Wrapper Class to Wrap the Python and C++ API into a combined python API

forward(shapes: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d358>, params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d390>)

Actual prediction

Parameters:

• shapes (torch.Tensor) – shapes before applied global transformation
• params (torch.Tensor) – parameters specifying the global transformation

Returns:

Transformed shapes

Return type:

torch.Tensor

num_params

_HomogeneousTransformationLayerPy

class _HomogeneousTransformationLayerPy(n_dims)

Bases: sphinx.ext.autodoc.importer._MockObject

Module to perform homogeneous transformations in 2D and 3D (Implemented in Python)

_ensemble_2d_matrix(rotation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d710>, translation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d748>, scale_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d780>)

ensembles the homogeneous transformation matrix for 2D

Parameters:

• rotation_params (torch.Tensor) – parameters specifying the rotation (one parameter)
• translation_params (torch.Tensor) – parameters specifying the translation (two parameters)
• scale_params (torch.Tensor) – parameter specifying the global scaling factor (one parameter) (currently only isotropic scaling supported)

Returns:

2D transformation matrix

Return type:

torch.Tensor

_ensemble_3d_matrix(rotation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d7b8>, translation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d7f0>, scale_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d828>)

ensembles the homogeneous transformation matrix for 3D

Parameters:

• rotation_params (torch.Tensor) – parameters specifying the rotation (three parameters)
• translation_params (torch.Tensor) – parameters specifying the translation (three parameters)
• scale_params (torch.Tensor) – parameter specifying the global scaling factor (one parameter) (currently only isotropic scaling supported)

Returns:

3D transformation matrix

Return type:

torch.Tensor

_ensemble_trafo(rotation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d668>, translation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d6a0>, scale_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d6d8>)

ensembles the transformation matrix in 2D and 3D

Parameters:

• rotation_params (torch.Tensor) – parameters specifying the rotation (one per DoF)
• translation_params (torch.Tensor) – parameters specifying the translation (one per dimension)
• scale_params (torch.Tensor) – parameter specifying the global scaling factor (currently only isotropic scaling supported)

Returns:

transformation matrix

Return type:

torch.Tensor

forward(shapes: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d588>, rotation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d5c0>, translation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d5f8>, scale_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d630>)

ensembles the homogeneous transformation matrix and applies it to the shape tensor

Parameters:

• shapes (torch.Tensor) – shapes to transform
• rotation_params (torch.Tensor) – parameters specifying the rotation (one per DoF)
• translation_params (torch.Tensor) – parameters specifying the translation (one per dimension)
• scale_params (torch.Tensor) – parameter specifying the global scaling factor (currently only isotropic scaling supported)

Returns:

the transformed shapes in cartesian coordinates

Return type:

torch.Tensor

_HomogeneousTransformationLayerCpp

class _HomogeneousTransformationLayerCpp(n_dims, verbose=True)

Bases: sphinx.ext.autodoc.importer._MockObject

Module to perform homogeneous transformations in 2D and 3D (Implemented in C++)

forward(shapes: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d438>, rotation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d470>, translation_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d4a8>, scale_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843a2d4e0>)

ensembles the homogeneous transformation matrix and applies it to the shape tensor

Parameters:

• shapes (torch.Tensor) – shapes to transform
• rotation_params (torch.Tensor) – parameters specifying the rotation (one per DoF)
• translation_params (torch.Tensor) – parameters specifying the translation (one per dimension)
• scale_params (torch.Tensor) – parameter specifying the global scaling factor (currently only isotropic scaling supported)

Returns:

the transformed shapes in cartesian coordinates

Return type:

torch.Tensor