HomogeneousTransformationLayer¶
-
class
HomogeneousTransformationLayer
(n_dims: int, use_cpp=False)[source]¶ 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>)[source]¶ 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
- shapes (
-
num_params
¶
-
_HomogeneousTransformationLayerPy¶
-
class
_HomogeneousTransformationLayerPy
(n_dims)[source]¶ 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>)[source]¶ 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
- rotation_params (
-
_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>)[source]¶ 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
- rotation_params (
-
_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>)[source]¶ 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
- rotation_params (
-
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>)[source]¶ 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
- shapes (
-
_HomogeneousTransformationLayerCpp¶
-
class
_HomogeneousTransformationLayerCpp
(n_dims, verbose=True)[source]¶ 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>)[source]¶ 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
- shapes (
-