kornia.geometry

Geometric image transformations is another key ingredient in computer vision to manipulate images. Since geometry operations are typically performed in 2D or 3D, we provide several algorithms to work with both cases. This module, the original core of the library, consists of the following submodules: transforms, camera, conversions, linalg and depth. We next describe each of them:

• transforms: The module provides low level interfaces to manipulate 2D images, with routines for Rotating, Scaling, Translating, Shearing; Cropping functions in several modalities such as central crops, crop and resize; Flipping transformations in the vertical and horizontal axis; Resizing operations; Functions to warp tensors given affine or perspective transformations, and utilities to compute the transformation matrices to perform the mentioned operations.

• camera: A set of routines specific to different types of camera representations such as Pinhole or Orthographic models containing functionalities such as projecting and unprojecting points from the camera to a world frame.

• conversions: Routines to perform conversions between angle representation such as radians to degrees, coordinates normalization, and homogeneous to euclidean. Moreover, we include advanced conversions for 3D geometry representations such as Quaternion, Axis-Angle, Rotation Matrix, or Rodrigues formula.

• linalg: Functions to perform general rigid-body homogeneous transformations. We include implementations to transform points between frames and for homogeneous transformations, manipulation such as composition, inverse and to compute relative poses.

• depth: A set of layers to manipulate depth maps such as how to compute 3D point clouds given depth maps and calibrated cameras; compute surface normals per pixel and warp tensor frames given calibrated cameras setup.

Resources:

align_corners

align_corners is a switch that widely offered in PyTorch geometric transform functions. Here is a simple illustration showing how a 4x4 image is upsampled to 8x8, made by bkkm16.

• align_corners=True, pixels are arranged as a grid of points. Points at the corners are aligned.

• align_corners=False, pixels are arranged as 1x1 areas. Area boundaries, rather than their centers, are aligned.

• kornia.geometry.bbox
  • bbox_generator()
  • bbox_generator3d()
  • bbox_to_mask()
  • bbox_to_mask3d()
  • infer_bbox_shape()
  • infer_bbox_shape3d()
  • nms()
  • transform_bbox()
  • validate_bbox()
  • validate_bbox3d()
• kornia.geometry.boxes
  • Boxes
    • Boxes.compute_area()
    • Boxes.device
    • Boxes.dtype
    • Boxes.from_tensor()
    • Boxes.get_boxes_shape()
    • Boxes.merge()
    • Boxes.pad()
    • Boxes.to()
    • Boxes.to_mask()
    • Boxes.to_tensor()
    • Boxes.transform_boxes()
    • Boxes.transform_boxes_()
    • Boxes.translate()
    • Boxes.trim()
    • Boxes.unpad()
  • Boxes3D
    • Boxes3D.device
    • Boxes3D.dtype
    • Boxes3D.from_tensor()
    • Boxes3D.get_boxes_shape()
    • Boxes3D.to()
    • Boxes3D.to_mask()
    • Boxes3D.to_tensor()
    • Boxes3D.transform_boxes()
    • Boxes3D.transform_boxes_()
• kornia.geometry.keypoints
  • Keypoints
    • Keypoints.clone()
    • Keypoints.data
    • Keypoints.device
    • Keypoints.dtype
    • Keypoints.from_tensor()
    • Keypoints.index_put()
    • Keypoints.pad()
    • Keypoints.shape
    • Keypoints.to_tensor()
    • Keypoints.transform_keypoints()
    • Keypoints.transform_keypoints_()
    • Keypoints.type()
    • Keypoints.unpad()
  • Keypoints3D
    • Keypoints3D.clone()
    • Keypoints3D.data
    • Keypoints3D.from_tensor()
    • Keypoints3D.pad()
    • Keypoints3D.shape
    • Keypoints3D.to_tensor()
    • Keypoints3D.transform_keypoints()
    • Keypoints3D.transform_keypoints_()
    • Keypoints3D.unpad()
• kornia.geometry.calibration
  • Undistortion
    • undistort_image()
    • undistort_points()
    • distort_points()
    • tilt_projection()
  • Perspective-n-Point (PnP)
    • solve_pnp_dlt()
• kornia.geometry.camera
  • Pinhole Camera
    • PinholeCamera
    • cam2pixel()
    • pixel2cam()
  • Perspective Camera
    • project_points()
    • unproject_points()
  • Stereo Camera
    • StereoCamera
    • reproject_disparity_to_3D()
• kornia.geometry.conversions
  • Angles
    • rad2deg()
    • deg2rad()
    • pol2cart()
    • cart2pol()
    • angle_to_rotation_matrix()
  • Coordinates
    • convert_points_from_homogeneous()
    • convert_points_to_homogeneous()
    • convert_affinematrix_to_homography()
    • denormalize_pixel_coordinates()
    • normalize_pixel_coordinates()
    • denormalize_pixel_coordinates3d()
    • normalize_pixel_coordinates3d()
    • normalize_points_with_intrinsics()
    • denormalize_points_with_intrinsics()
  • Homography
    • normalize_homography()
    • denormalize_homography()
    • normalize_homography3d()
  • Quaternion
    • quaternion_to_angle_axis()
    • quaternion_to_rotation_matrix()
    • quaternion_log_to_exp()
    • quaternion_exp_to_log()
    • normalize_quaternion()
  • Rotation Matrix
    • rotation_matrix_to_angle_axis()
    • rotation_matrix_to_quaternion()
  • Angle Axis
    • angle_axis_to_quaternion()
    • angle_axis_to_rotation_matrix()
  • Euler Angles
    • quaternion_from_euler()
    • euler_from_quaternion()
  • Pose
    • Rt_to_matrix4x4()
    • matrix4x4_to_Rt()
    • worldtocam_to_camtoworld_Rt()
    • camtoworld_to_worldtocam_Rt()
    • camtoworld_graphics_to_vision_4x4()
    • camtoworld_vision_to_graphics_4x4()
    • camtoworld_graphics_to_vision_Rt()
    • camtoworld_vision_to_graphics_Rt()
    • ARKitQTVecs_to_ColmapQTVecs()
• kornia.geometry.depth
  • depth_from_disparity()
  • depth_to_3d()
  • depth_to_normals()
  • warp_frame_depth()
• kornia.geometry.epipolar
  • Essential
    • essential_from_fundamental()
    • essential_from_Rt()
    • decompose_essential_matrix()
    • motion_from_essential()
    • motion_from_essential_choose_solution()
    • relative_camera_motion()
  • Fundamental
    • find_fundamental()
    • fundamental_from_essential()
    • fundamental_from_projections()
    • compute_correspond_epilines()
    • normalize_points()
    • normalize_transformation()
    • get_perpendicular()
    • get_closest_point_on_epipolar_line()
  • Metrics
    • sampson_epipolar_distance()
    • symmetrical_epipolar_distance()
    • left_to_right_epipolar_distance()
    • right_to_left_epipolar_distance()
  • Projection
    • projection_from_KRt()
    • projections_from_fundamental()
    • intrinsics_like()
    • scale_intrinsics()
    • random_intrinsics()
  • Numeric
    • cross_product_matrix()
  • Triangulation
    • triangulate_points()
• kornia.geometry.homography
  • find_homography_dlt()
  • find_homography_dlt_iterated()
  • find_homography_lines_dlt()
  • find_homography_lines_dlt_iterated()
  • line_segment_transfer_error_one_way()
  • oneway_transfer_error()
  • sample_is_valid_for_homography()
  • symmetric_transfer_error()
  • Interactive Demo
• kornia.geometry.liegroup
  • So3
    • So3.__init__()
    • So3.__mul__()
    • So3.__repr__()
    • So3.adjoint()
    • So3.exp()
    • So3.from_matrix()
    • So3.from_wxyz()
    • So3.hat()
    • So3.identity()
    • So3.inverse()
    • So3.log()
    • So3.matrix()
    • So3.q
    • So3.random()
    • So3.rot_x()
    • So3.rot_y()
    • So3.rot_z()
    • So3.vee()
  • Se3
    • Se3.__init__()
    • Se3.__mul__()
    • Se3.__repr__()
    • Se3.adjoint()
    • Se3.exp()
    • Se3.from_qxyz()
    • Se3.hat()
    • Se3.identity()
    • Se3.inverse()
    • Se3.log()
    • Se3.matrix()
    • Se3.r
    • Se3.random()
    • Se3.rot_x()
    • Se3.rot_y()
    • Se3.rot_z()
    • Se3.rotation
    • Se3.so3
    • Se3.t
    • Se3.trans()
    • Se3.trans_x()
    • Se3.trans_y()
    • Se3.trans_z()
    • Se3.translation
    • Se3.vee()
  • So2
    • So2.__init__()
    • So2.__mul__()
    • So2.__repr__()
    • So2.adjoint()
    • So2.exp()
    • So2.from_matrix()
    • So2.hat()
    • So2.identity()
    • So2.inverse()
    • So2.log()
    • So2.matrix()
    • So2.random()
    • So2.vee()
    • So2.z
  • Se2
    • Se2.__init__()
    • Se2.__mul__()
    • Se2.__repr__()
    • Se2.adjoint()
    • Se2.exp()
    • Se2.hat()
    • Se2.identity()
    • Se2.inverse()
    • Se2.log()
    • Se2.matrix()
    • Se2.r
    • Se2.random()
    • Se2.rotation
    • Se2.so2
    • Se2.t
    • Se2.trans()
    • Se2.trans_x()
    • Se2.trans_y()
    • Se2.translation
    • Se2.vee()
• kornia.geometry.linalg
  • relative_transformation()
  • compose_transformations()
  • inverse_transformation()
  • transform_points()
  • point_line_distance()
  • squared_norm()
  • batched_dot_product()
  • euclidean_distance()
• kornia.geometry.line
  • ParametrizedLine
    • ParametrizedLine.__init__()
    • ParametrizedLine.dim()
    • ParametrizedLine.direction
    • ParametrizedLine.distance()
    • ParametrizedLine.intersect()
    • ParametrizedLine.origin
    • ParametrizedLine.point_at()
    • ParametrizedLine.projection()
    • ParametrizedLine.squared_distance()
    • ParametrizedLine.through()
  • fit_line()
  • Hyperplane
  • Interactive Demo
• kornia.geometry.quaternion
  • Quaternion
    • Quaternion.__add__()
    • Quaternion.__init__()
    • Quaternion.__neg__()
    • Quaternion.__pow__()
    • Quaternion.__repr__()
    • Quaternion.__sub__()
    • Quaternion.coeffs
    • Quaternion.data
    • Quaternion.from_axis_angle()
    • Quaternion.from_coeffs()
    • Quaternion.from_matrix()
    • Quaternion.identity()
    • Quaternion.matrix()
    • Quaternion.polar_angle
    • Quaternion.q
    • Quaternion.random()
    • Quaternion.real
    • Quaternion.scalar
    • Quaternion.shape
    • Quaternion.slerp()
    • Quaternion.vec
    • Quaternion.w
    • Quaternion.x
    • Quaternion.y
    • Quaternion.z
• kornia.geometry.subpix
  • Convolutional
    • conv_soft_argmax2d()
    • conv_soft_argmax3d()
    • conv_quad_interp3d()
  • Spatial
    • spatial_softmax2d()
    • spatial_expectation2d()
    • spatial_soft_argmax2d()
    • render_gaussian2d()
  • Non Maxima Suppression
    • nms2d()
    • nms3d()
  • Module
    • SpatialSoftArgmax2d
    • ConvSoftArgmax2d
    • ConvSoftArgmax3d
    • ConvQuadInterp3d
    • NonMaximaSuppression2d
    • NonMaximaSuppression3d
• kornia.geometry.transform
  • Warp operators
    • warp_perspective()
    • warp_perspective3d()
    • warp_affine()
    • warp_affine3d()
    • warp_image_tps()
    • warp_points_tps()
    • remap()
  • Image 2d transforms
    • affine()
    • rotate()
    • translate()
    • scale()
    • shear()
    • hflip()
    • vflip()
    • rot180()
    • resize()
    • rescale()
    • elastic_transform2d()
    • pyrdown()
    • pyrup()
    • build_pyramid()
    • build_laplacian_pyramid()
    • upscale_double()
  • Matrix transformations
    • get_perspective_transform()
    • get_perspective_transform3d()
    • get_projective_transform()
    • get_rotation_matrix2d()
    • get_shear_matrix2d()
    • get_shear_matrix3d()
    • get_affine_matrix2d()
    • get_affine_matrix3d()
    • invert_affine_transform()
    • projection_from_Rt()
    • get_tps_transform()
  • Crop operators
    • crop_by_indices()
    • crop_by_boxes()
    • center_crop()
    • crop_and_resize()
  • Module
    • Rotate
    • Translate
    • Scale
    • Shear
    • PyrDown
    • PyrUp
    • ScalePyramid
    • Hflip
    • Vflip
    • Rot180
    • Resize
    • Rescale
    • Affine
    • HomographyWarper
  • Image registration
    • BaseModel
    • Homography
    • ImageRegistrator
    • Similarity
  • Interactive Demo
• kornia.geometry.ransac
  • RANSAC
    • RANSAC.forward()