- Successfully Transpiling to TensorFlow: 87.28%
- Successfully Transpiling to JAX: 92.01%
- Successfully Transpiling to Numpy (functions only): 91.2%
- Total Tests Passing: 1412
- Total Tests Failing: 155
- Percent Tests Passing: 90.1%
kornia
augmentation
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.augmentation.RandomMosaic |  |
 |
|
| kornia.augmentation.RandomTransplantation | |
 |
|
| kornia.augmentation.CenterCrop3D | |
|
|
| kornia.augmentation.RandomAffine3D | |
|
|
| kornia.augmentation.RandomCrop3D | |
|
|
| kornia.augmentation.RandomDepthicalFlip3D | |
|
|
| kornia.augmentation.RandomHorizontalFlip3D | |
|
|
| kornia.augmentation.RandomRotation3D | |
|
|
| kornia.augmentation.RandomVerticalFlip3D | |
|
|
| kornia.augmentation.RandomEqualize3D | |
|
|
| kornia.augmentation.RandomMotionBlur3D | |
|
|
| kornia.augmentation.RandomTransplantation3D | |
|
|
| kornia.augmentation.Denormalize | |
|
|
| kornia.augmentation.Normalize | |
|
|
| kornia.augmentation.LongestMaxSize | |
|
|
| kornia.augmentation.Resize | |
|
|
| kornia.augmentation.SmallestMaxSize | |
|
|
| kornia.augmentation.RandomLinearCornerIllumination | |
|
|
| kornia.augmentation.RandomLinearIllumination | |
|
|
| kornia.augmentation.RandomMedianBlur | |
|
|
| kornia.augmentation.RandomMotionBlur | |
|
|
| kornia.augmentation.RandomPlanckianJitter | |
|
|
| kornia.augmentation.RandomPlasmaBrightness | |
|
|
| kornia.augmentation.RandomPlasmaContrast | |
|
|
| kornia.augmentation.RandomPlasmaShadow | |
|
|
| kornia.augmentation.RandomPosterize | |
|
|
| kornia.augmentation.RandomRain | |
|
|
| kornia.augmentation.RandomRGBShift | |
|
|
| kornia.augmentation.RandomSaltAndPepperNoise | |
|
|
| kornia.augmentation.RandomSaturation | |
|
|
| kornia.augmentation.RandomSharpness | |
|
|
| kornia.augmentation.RandomSnow | |
|
|
| kornia.augmentation.RandomSolarize | |
|
|
| kornia.augmentation.CenterCrop | |
|
|
| kornia.augmentation.container.AugmentationSequential | |
|
|
| kornia.augmentation.container.ManyToManyAugmentationDispather | |
|
|
| kornia.augmentation.container.ManyToOneAugmentationDispather | |
|
|
| kornia.augmentation.container.ImageSequential | |
|
|
| kornia.augmentation.container.PatchSequential | |
|
|
| kornia.augmentation.container.VideoSequential | |
|
|
| kornia.augmentation.ColorJiggle | |
|
|
| kornia.augmentation.ColorJitter | |
|
|
| kornia.augmentation.RandomAutoContrast | |
|
|
| kornia.augmentation.RandomBoxBlur | |
|
|
| kornia.augmentation.RandomBrightness | |
|
|
| kornia.augmentation.RandomChannelDropout | |
|
|
| kornia.augmentation.RandomChannelShuffle | |
|
|
| kornia.augmentation.RandomClahe | |
|
|
| kornia.augmentation.RandomContrast | |
|
|
| kornia.augmentation.RandomEqualize | |
|
|
| kornia.augmentation.RandomGamma | |
|
|
| kornia.augmentation.RandomGaussianBlur | |
|
|
| kornia.augmentation.RandomGaussianIllumination | |
|
|
| kornia.augmentation.RandomGaussianNoise | |
|
|
| kornia.augmentation.RandomGrayscale | |
|
|
| kornia.augmentation.RandomHue | |
|
|
| kornia.augmentation.RandomInvert | |
|
|
| kornia.augmentation.RandomJPEG | |
|
|
| kornia.augmentation.auto.AutoAugment | |
|
|
| kornia.augmentation.auto.RandAugment | |
|
|
| kornia.augmentation.auto.TrivialAugment | |
|
|
| kornia.augmentation.PadTo | |
|
|
| kornia.augmentation.RandomAffine | |
|
|
| kornia.augmentation.RandomCrop | |
|
|
| kornia.augmentation.RandomElasticTransform | |
|
|
| kornia.augmentation.RandomErasing | |
|
|
| kornia.augmentation.RandomFisheye | |
|
|
| kornia.augmentation.RandomHorizontalFlip | |
|
|
| kornia.augmentation.RandomPerspective | |
|
|
| kornia.augmentation.RandomResizedCrop | |
|
|
| kornia.augmentation.RandomRotation | |
|
|
| kornia.augmentation.RandomShear | |
|
|
| kornia.augmentation.RandomThinPlateSpline | |
|
|
| kornia.augmentation.RandomVerticalFlip | |
|
|
| kornia.augmentation.RandomCutMixV2 | |
|
|
| kornia.augmentation.RandomJigsaw | |
|
|
| kornia.augmentation.RandomMixUpV2 | |
|
|
color
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.color.gray.rgb_to_grayscale | |
|
|
| kornia.color.gray.bgr_to_grayscale | |
|
|
| kornia.color.gray.grayscale_to_rgb | |
|
|
| kornia.color.rgb.rgb_to_bgr | |
|
|
| kornia.color.rgb.bgr_to_rgb | |
|
|
| kornia.color.rgb.rgb_to_linear_rgb | |
|
|
| kornia.color.rgb.linear_rgb_to_rgb | |
|
|
| kornia.color.rgb.bgr_to_rgba | |
|
|
| kornia.color.rgb.rgb_to_rgba | |
|
|
| kornia.color.rgb.rgba_to_rgb | |
|
|
| kornia.color.rgb.rgba_to_bgr | |
|
|
| kornia.color.hls.rgb_to_hls | |
|
|
| kornia.color.hls.hls_to_rgb | |
|
|
| kornia.color.hsv.rgb_to_hsv | |
|
|
| kornia.color.hsv.hsv_to_rgb | |
|
|
| kornia.color.luv.rgb_to_luv | |
|
|
| kornia.color.luv.luv_to_rgb | |
|
|
| kornia.color.lab.rgb_to_lab | |
|
|
| kornia.color.lab.lab_to_rgb | |
|
|
| kornia.color.ycbcr.rgb_to_ycbcr | |
|
|
| kornia.color.ycbcr.ycbcr_to_rgb | |
|
|
| kornia.color.yuv.rgb_to_yuv | |
|
|
| kornia.color.yuv.yuv_to_rgb | |
|
|
| kornia.color.yuv.rgb_to_yuv420 | |
|
|
| kornia.color.yuv.yuv420_to_rgb | |
|
|
| kornia.color.yuv.rgb_to_yuv422 | |
|
|
| kornia.color.yuv.yuv422_to_rgb | |
|
|
| kornia.color.xyz.rgb_to_xyz | |
|
|
| kornia.color.xyz.xyz_to_rgb | |
|
|
| kornia.color.rgb_to_raw | |
|
|
| kornia.color.raw_to_rgb | |
|
|
| kornia.color.raw_to_rgb_2x2_downscaled | |
|
|
| kornia.color.sepia.sepia_from_rgb | |
|
|
| kornia.color.gray.GrayscaleToRgb | |
|
|
| kornia.color.gray.RgbToGrayscale | |
|
|
| kornia.color.gray.BgrToGrayscale | |
|
|
| kornia.color.rgb.RgbToBgr | |
|
|
| kornia.color.rgb.BgrToRgb | |
|
|
| kornia.color.rgb.LinearRgbToRgb | |
|
|
| kornia.color.rgb.RgbToLinearRgb | |
|
|
| kornia.color.rgb.RgbToRgba | |
|
|
| kornia.color.rgb.BgrToRgba | |
|
|
| kornia.color.rgb.RgbaToRgb | |
|
|
| kornia.color.rgb.RgbaToBgr | |
|
|
| kornia.color.hls.RgbToHls | |
|
|
| kornia.color.hls.HlsToRgb | |
|
|
| kornia.color.hsv.RgbToHsv | |
|
|
| kornia.color.hsv.HsvToRgb | |
|
|
| kornia.color.luv.RgbToLuv | |
|
|
| kornia.color.luv.LuvToRgb | |
|
|
| kornia.color.lab.RgbToLab | |
|
|
| kornia.color.lab.LabToRgb | |
|
|
| kornia.color.ycbcr.YcbcrToRgb | |
|
|
| kornia.color.ycbcr.RgbToYcbcr | |
|
|
| kornia.color.yuv.RgbToYuv | |
|
|
| kornia.color.yuv.YuvToRgb | |
|
|
| kornia.color.yuv.RgbToYuv420 | |
|
|
| kornia.color.yuv.Yuv420ToRgb | |
|
|
| kornia.color.yuv.RgbToYuv422 | |
|
|
| kornia.color.yuv.Yuv422ToRgb | |
|
|
| kornia.color.xyz.RgbToXyz | |
|
|
| kornia.color.xyz.XyzToRgb | |
|
|
| kornia.color.RawToRgb | |
|
|
| kornia.color.RgbToRaw | |
|
|
| kornia.color.RawToRgb2x2Downscaled | |
|
|
| kornia.color.sepia.Sepia | |
|
|
| kornia.color.CFA | |
|
|
| kornia.color.ColorMap | |
|
|
contrib
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.contrib.extract_patches.compute_padding | |
|
|
| kornia.contrib.extract_patches.extract_tensor_patches | |
|
|
| kornia.contrib.extract_patches.combine_tensor_patches | |
|
|
| kornia.contrib.distance_transform.distance_transform | |
|
|
| kornia.contrib.diamond_square.diamond_square | |
|
|
| kornia.contrib.EdgeDetector | |
|
|
| kornia.contrib.FaceDetector | |
|
|
| kornia.contrib.VisionTransformer | |
|
|
| kornia.contrib.KMeans | |
|
|
| kornia.contrib.ExtractTensorPatches | |
|
|
| kornia.contrib.CombineTensorPatches | |
|
|
| kornia.contrib.ClassificationHead | |
|
|
| kornia.contrib.ImageStitcher | |
|
|
| kornia.contrib.Lambda | |
|
|
| kornia.contrib.DistanceTransform | |
|
|
enhance
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.enhance.core.add_weighted | |
|
|
| kornia.enhance.adjust.adjust_brightness | |
|
|
| kornia.enhance.adjust.adjust_contrast | |
|
|
| kornia.enhance.adjust.adjust_contrast_with_mean_subtraction | |
|
|
| kornia.enhance.adjust.adjust_gamma | |
|
|
| kornia.enhance.adjust.adjust_hue | |
|
|
| kornia.enhance.adjust.adjust_saturation | |
|
|
| kornia.enhance.adjust.adjust_sigmoid | |
|
|
| kornia.enhance.adjust.adjust_log | |
|
|
| kornia.enhance.adjust.invert | |
|
|
| kornia.enhance.adjust.posterize | |
|
|
| kornia.enhance.adjust.sharpness | |
|
|
| kornia.enhance.adjust.solarize | |
|
|
| kornia.enhance.adjust.equalize | |
|
|
| kornia.enhance.equalization.equalize_clahe | |
|
|
| kornia.enhance.adjust.equalize3d | |
|
|
| kornia.enhance.histogram.histogram | |
|
|
| kornia.enhance.histogram.histogram2d | |
|
|
| kornia.enhance.histogram.image_histogram2d | |
|
|
| kornia.enhance.normalize.normalize | |
|
|
| kornia.enhance.normalize.normalize_min_max | |
|
|
| kornia.enhance.normalize.denormalize | |
|
|
| kornia.enhance.zca.zca_mean | |
|
|
| kornia.enhance.zca.zca_whiten | |
|
|
| kornia.enhance.zca.linear_transform | |
|
|
| kornia.enhance.jpeg.jpeg_codec_differentiable | |
|
|
| kornia.enhance.Normalize | |
|
|
| kornia.enhance.Denormalize | |
|
|
| kornia.enhance.ZCAWhitening | |
|
|
| kornia.enhance.AdjustBrightness | |
|
|
| kornia.enhance.AdjustContrast | |
|
|
| kornia.enhance.AdjustSaturation | |
|
|
| kornia.enhance.AdjustHue | |
|
|
| kornia.enhance.AdjustGamma | |
|
|
| kornia.enhance.AdjustSigmoid | |
|
|
| kornia.enhance.AdjustLog | |
|
|
| kornia.enhance.AddWeighted | |
|
|
| kornia.enhance.Invert | |
|
|
| kornia.enhance.JPEGCodecDifferentiable | |
|
|
feature
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.feature.responses.gftt_response | |
|
|
| kornia.feature.responses.harris_response | |
|
|
| kornia.feature.responses.hessian_response | |
|
|
| kornia.feature.responses.dog_response | |
|
|
| kornia.feature.responses.dog_response_single | |
|
|
| kornia.feature.integrated.get_laf_descriptors | |
|
|
| kornia.feature.matching.match_nn | |
|
|
| kornia.feature.matching.match_mnn | |
|
|
| kornia.feature.matching.match_snn | |
|
|
| kornia.feature.matching.match_smnn | |
|
|
| kornia.feature.matching.match_fginn | |
|
|
| kornia.feature.adalam.adalam.match_adalam | |
|
|
| kornia.feature.laf.extract_patches_from_pyramid | |
|
|
| kornia.feature.laf.extract_patches_simple | |
|
|
| kornia.feature.laf.normalize_laf | |
|
|
| kornia.feature.laf.denormalize_laf | |
|
|
| kornia.feature.laf.laf_to_boundary_points | |
|
|
| kornia.feature.laf.ellipse_to_laf | |
|
|
| kornia.feature.laf.make_upright | |
|
|
| kornia.feature.TFeat | |
|
|
| kornia.feature.SOSNet | |
|
|
| kornia.feature.BlobHessian | |
|
|
| kornia.feature.CornerGFTT | |
|
|
| kornia.feature.CornerHarris | |
|
|
| kornia.feature.BlobDoG | |
|
|
| kornia.feature.BlobDoGSingle | |
|
|
| kornia.feature.KeyNet | |
|
|
| kornia.feature.MultiResolutionDetector | |
|
|
| kornia.feature.ScaleSpaceDetector | |
|
|
| kornia.feature.KeyNetDetector | |
|
|
| kornia.feature.LAFDescriptor | |
|
|
| kornia.feature.LocalFeature | |
|
|
| kornia.feature.SIFTFeature | |
|
|
| kornia.feature.SIFTFeatureScaleSpace | |
|
|
| kornia.feature.GFTTAffNetHardNet | |
|
|
| kornia.feature.KeyNetAffNetHardNet | |
|
|
| kornia.feature.KeyNetHardNet | |
|
|
| kornia.feature.DescriptorMatcher | |
|
|
| kornia.feature.GeometryAwareDescriptorMatcher | |
|
|
| kornia.feature.LocalFeatureMatcher | |
|
|
| kornia.feature.LightGlueMatcher | |
|
|
| kornia.feature.LightGlue | |
|
|
| kornia.feature.LoFTR | |
|
|
| kornia.feature.PassLAF | |
|
|
| kornia.feature.PatchAffineShapeEstimator | |
|
|
| kornia.feature.LAFAffineShapeEstimator | |
|
|
| kornia.feature.laf.scale_laf | |
|
|
| kornia.feature.laf.get_laf_scale | |
|
|
| kornia.feature.laf.get_laf_center | |
|
|
| kornia.feature.laf.rotate_laf | |
|
|
| kornia.feature.laf.get_laf_orientation | |
|
|
| kornia.feature.laf.set_laf_orientation | |
|
|
| kornia.feature.laf.laf_from_center_scale_ori | |
|
|
| kornia.feature.laf.laf_is_inside_image | |
|
|
| kornia.feature.laf.laf_to_three_points | |
|
|
| kornia.feature.laf.laf_from_three_points | |
|
|
| kornia.feature.laf.perspective_transform_lafs | |
|
|
| kornia.feature.DenseSIFTDescriptor | |
|
|
| kornia.feature.SIFTDescriptor | |
|
|
| kornia.feature.MKDDescriptor | |
|
|
| kornia.feature.HardNet | |
|
|
| kornia.feature.HardNet8 | |
|
|
| kornia.feature.HyNet | |
|
|
| kornia.feature.LAFOrienter | |
|
|
| kornia.feature.PatchDominantGradientOrientation | |
|
|
| kornia.feature.OriNet | |
|
|
| kornia.feature.LAFAffNetShapeEstimator | |
|
|
| kornia.feature.FilterResponseNorm2d | |
|
|
| kornia.feature.TLU | |
|
|
| kornia.feature.DeFMO | |
|
|
| kornia.feature.DeDoDe | |
|
|
| kornia.feature.DISK | |
|
|
| kornia.feature.DISKFeatures | |
|
|
filters
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.filters.bilateral.bilateral_blur | |
|
|
| kornia.filters.blur_pool.blur_pool2d | |
|
|
| kornia.filters.blur.box_blur | |
|
|
| kornia.filters.gaussian.gaussian_blur2d | |
|
|
| kornia.filters.guided.guided_blur | |
|
|
| kornia.filters.bilateral.joint_bilateral_blur | |
|
|
| kornia.filters.blur_pool.max_blur_pool2d | |
|
|
| kornia.filters.median.median_blur | |
|
|
| kornia.filters.motion.motion_blur | |
|
|
| kornia.filters.unsharp.unsharp_mask | |
|
|
| kornia.filters.canny.canny | |
|
|
| kornia.filters.laplacian.laplacian | |
|
|
| kornia.filters.sobel.sobel | |
|
|
| kornia.filters.sobel.spatial_gradient | |
|
|
| kornia.filters.sobel.spatial_gradient3d | |
|
|
| kornia.filters.filter.filter2d | |
|
|
| kornia.filters.filter.filter2d_separable | |
|
|
| kornia.filters.filter.filter3d | |
|
|
| kornia.filters.kernels.get_gaussian_kernel1d | |
|
|
| kornia.filters.kernels.get_gaussian_erf_kernel1d | |
|
|
| kornia.filters.kernels.get_gaussian_discrete_kernel1d | |
|
|
| kornia.filters.kernels.get_gaussian_kernel2d | |
|
|
| kornia.filters.kernels.get_hanning_kernel1d | |
|
|
| kornia.filters.kernels.get_hanning_kernel2d | |
|
|
| kornia.filters.kernels.get_laplacian_kernel1d | |
|
|
| kornia.filters.kernels.get_laplacian_kernel2d | |
|
|
| kornia.filters.kernels_geometry.get_motion_kernel2d | |
|
|
| kornia.filters.Laplacian | |
|
|
| kornia.filters.Sobel | |
|
|
| kornia.filters.Canny | |
|
|
| kornia.filters.SpatialGradient | |
|
|
| kornia.filters.SpatialGradient3d | |
|
|
| kornia.filters.DexiNed | |
|
|
| kornia.filters.BilateralBlur | |
|
|
| kornia.filters.BlurPool2D | |
|
|
| kornia.filters.BoxBlur | |
|
|
| kornia.filters.MaxBlurPool2D | |
|
|
| kornia.filters.MedianBlur | |
|
|
| kornia.filters.GaussianBlur2d | |
|
|
| kornia.filters.GuidedBlur | |
|
|
| kornia.filters.JointBilateralBlur | |
|
|
| kornia.filters.MotionBlur | |
|
|
| kornia.filters.UnsharpMask | |
|
|
| kornia.filters.InRange | |
|
|
geometry.bbox
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.bbox.bbox_generator | |
|
|
| kornia.geometry.bbox.bbox_generator3d | |
|
|
| kornia.geometry.bbox.bbox_to_mask | |
|
|
| kornia.geometry.bbox.bbox_to_mask3d | |
|
|
| kornia.geometry.bbox.infer_bbox_shape | |
|
|
| kornia.geometry.bbox.infer_bbox_shape3d | |
|
|
| kornia.geometry.bbox.nms | |
|
|
| kornia.geometry.bbox.transform_bbox | |
|
|
geometry.boxes
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.boxes.Boxes | |
|
|
| kornia.geometry.boxes.Boxes3D | |
|
|
geometry.calibration
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.calibration.undistort.undistort_image | |
|
|
| kornia.geometry.calibration.undistort.undistort_points | |
|
|
| kornia.geometry.calibration.distort.distort_points | |
|
|
| kornia.geometry.calibration.distort.tilt_projection | |
|
|
| kornia.geometry.calibration.pnp.solve_pnp_dlt | |
|
|
geometry.camera
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.camera.projection_z1.project_points_z1 | |
|
|
| kornia.geometry.camera.projection_z1.unproject_points_z1 | |
|
|
| kornia.geometry.camera.projection_z1.dx_project_points_z1 | |
|
|
| kornia.geometry.camera.projection_orthographic.project_points_orthographic | |
|
|
| kornia.geometry.camera.projection_orthographic.unproject_points_orthographic | |
|
|
| kornia.geometry.camera.projection_orthographic.dx_project_points_orthographic | |
|
|
| kornia.geometry.camera.distortion_affine.distort_points_affine | |
|
|
| kornia.geometry.camera.distortion_affine.undistort_points_affine | |
|
|
| kornia.geometry.camera.distortion_affine.dx_distort_points_affine | |
|
|
| kornia.geometry.camera.distortion_kannala_brandt.distort_points_kannala_brandt | |
|
|
| kornia.geometry.camera.distortion_kannala_brandt.undistort_points_kannala_brandt | |
|
|
| kornia.geometry.camera.distortion_kannala_brandt.dx_distort_points_kannala_brandt | |
|
|
| kornia.geometry.camera.pinhole.cam2pixel | |
|
|
| kornia.geometry.camera.pinhole.pixel2cam | |
|
|
| kornia.geometry.camera.perspective.project_points | |
|
|
| kornia.geometry.camera.perspective.unproject_points | |
|
|
| kornia.geometry.camera.stereo.reproject_disparity_to_3D | |
|
|
| kornia.geometry.camera.pinhole.PinholeCamera | |
|
|
| kornia.geometry.camera.stereo.StereoCamera | |
|
|
geometry.conversions
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.conversions.rad2deg | |
|
|
| kornia.geometry.conversions.deg2rad | |
|
|
| kornia.geometry.conversions.pol2cart | |
|
|
| kornia.geometry.conversions.cart2pol | |
|
|
| kornia.geometry.conversions.angle_to_rotation_matrix | |
|
|
| kornia.geometry.conversions.convert_points_from_homogeneous | |
|
|
| kornia.geometry.conversions.convert_points_to_homogeneous | |
|
|
| kornia.geometry.conversions.convert_affinematrix_to_homography | |
|
|
| kornia.geometry.conversions.denormalize_pixel_coordinates | |
|
|
| kornia.geometry.conversions.normalize_pixel_coordinates | |
|
|
| kornia.geometry.conversions.denormalize_pixel_coordinates3d | |
|
|
| kornia.geometry.conversions.normalize_pixel_coordinates3d | |
|
|
| kornia.geometry.conversions.normalize_points_with_intrinsics | |
|
|
| kornia.geometry.conversions.denormalize_points_with_intrinsics | |
|
|
| kornia.geometry.conversions.normalize_homography | |
|
|
| kornia.geometry.conversions.denormalize_homography | |
|
|
| kornia.geometry.conversions.normalize_homography3d | |
|
|
| kornia.geometry.conversions.quaternion_to_axis_angle | |
|
|
| kornia.geometry.conversions.quaternion_to_rotation_matrix | |
|
|
| kornia.geometry.conversions.quaternion_log_to_exp | |
|
|
| kornia.geometry.conversions.quaternion_exp_to_log | |
|
|
| kornia.geometry.conversions.normalize_quaternion | |
|
|
| kornia.geometry.conversions.vector_to_skew_symmetric_matrix | |
|
|
| kornia.geometry.conversions.rotation_matrix_to_axis_angle | |
|
|
| kornia.geometry.conversions.rotation_matrix_to_quaternion | |
|
|
| kornia.geometry.conversions.axis_angle_to_quaternion | |
|
|
| kornia.geometry.conversions.axis_angle_to_rotation_matrix | |
|
|
| kornia.geometry.conversions.quaternion_from_euler | |
|
|
| kornia.geometry.conversions.euler_from_quaternion | |
|
|
| kornia.geometry.conversions.Rt_to_matrix4x4 | |
|
|
| kornia.geometry.conversions.matrix4x4_to_Rt | |
|
|
| kornia.geometry.conversions.worldtocam_to_camtoworld_Rt | |
|
|
| kornia.geometry.conversions.camtoworld_to_worldtocam_Rt | |
|
|
| kornia.geometry.conversions.camtoworld_graphics_to_vision_4x4 | |
|
|
| kornia.geometry.conversions.camtoworld_vision_to_graphics_4x4 | |
|
|
| kornia.geometry.conversions.camtoworld_graphics_to_vision_Rt | |
|
|
| kornia.geometry.conversions.camtoworld_vision_to_graphics_Rt | |
|
|
| kornia.geometry.conversions.ARKitQTVecs_to_ColmapQTVecs | |
|
|
geometry.depth
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.depth.depth_from_disparity | |
|
|
| kornia.geometry.depth.depth_to_3d | |
|
|
| kornia.geometry.depth.depth_to_3d_v2 | |
|
|
| kornia.geometry.depth.unproject_meshgrid | |
|
|
| kornia.geometry.depth.depth_to_normals | |
|
|
| kornia.geometry.depth.warp_frame_depth | |
|
|
geometry.epipolar
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.epipolar.essential.find_essential | |
|
|
| kornia.geometry.epipolar.essential.essential_from_fundamental | |
|
|
| kornia.geometry.epipolar.essential.essential_from_Rt | |
|
|
| kornia.geometry.epipolar.essential.decompose_essential_matrix | |
|
|
| kornia.geometry.epipolar.essential.motion_from_essential | |
|
|
| kornia.geometry.epipolar.essential.motion_from_essential_choose_solution | |
|
|
| kornia.geometry.epipolar.essential.relative_camera_motion | |
|
|
| kornia.geometry.epipolar.fundamental.find_fundamental | |
|
|
| kornia.geometry.epipolar.fundamental.fundamental_from_essential | |
|
|
| kornia.geometry.epipolar.fundamental.fundamental_from_projections | |
|
|
| kornia.geometry.epipolar.fundamental.compute_correspond_epilines | |
|
|
| kornia.geometry.epipolar.fundamental.normalize_points | |
|
|
| kornia.geometry.epipolar.fundamental.normalize_transformation | |
|
|
| kornia.geometry.epipolar.fundamental.get_perpendicular | |
|
|
| kornia.geometry.epipolar.fundamental.get_closest_point_on_epipolar_line | |
|
|
| kornia.geometry.epipolar._metrics.sampson_epipolar_distance | |
|
|
| kornia.geometry.epipolar._metrics.symmetrical_epipolar_distance | |
|
|
| kornia.geometry.epipolar._metrics.left_to_right_epipolar_distance | |
|
|
| kornia.geometry.epipolar._metrics.right_to_left_epipolar_distance | |
|
|
| kornia.geometry.epipolar.projection.projection_from_KRt | |
|
|
| kornia.geometry.epipolar.projection.projections_from_fundamental | |
|
|
| kornia.geometry.epipolar.projection.intrinsics_like | |
|
|
| kornia.geometry.epipolar.projection.scale_intrinsics | |
|
|
| kornia.geometry.epipolar.projection.random_intrinsics | |
|
|
| kornia.geometry.epipolar.numeric.cross_product_matrix | |
|
|
| kornia.geometry.epipolar.triangulation.triangulate_points | |
|
|
geometry.homography
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.homography.find_homography_dlt | |
|
|
| kornia.geometry.homography.find_homography_dlt_iterated | |
|
|
| kornia.geometry.homography.find_homography_lines_dlt | |
|
|
| kornia.geometry.homography.find_homography_lines_dlt_iterated | |
|
|
| kornia.geometry.homography.line_segment_transfer_error_one_way | |
|
|
| kornia.geometry.homography.oneway_transfer_error | |
|
|
| kornia.geometry.homography.sample_is_valid_for_homography | |
|
|
| kornia.geometry.homography.symmetric_transfer_error | |
|
|
geometry.liegroup
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.liegroup.So3 | |
|
|
| kornia.geometry.liegroup.Se3 | |
|
|
| kornia.geometry.liegroup.So2 | |
|
|
| kornia.geometry.liegroup.Se2 | |
|
|
geometry.linalg
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.linalg.relative_transformation | |
|
|
| kornia.geometry.linalg.compose_transformations | |
|
|
| kornia.geometry.linalg.inverse_transformation | |
|
|
| kornia.geometry.linalg.transform_points | |
|
|
| kornia.geometry.linalg.point_line_distance | |
|
|
| kornia.geometry.linalg.batched_squared_norm | |
|
|
| kornia.geometry.linalg.batched_dot_product | |
|
|
| kornia.geometry.linalg.euclidean_distance | |
|
|
geometry.line
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.line.fit_line | |
|
|
| kornia.geometry.line.ParametrizedLine | |
|
|
geometry.quaternion
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.quaternion.Quaternion | |
|
|
geometry.ransac
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.ransac.RANSAC | |
|
|
geometry.solvers
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.solvers.polynomial_solver.solve_quadratic | |
|
|
| kornia.geometry.solvers.polynomial_solver.solve_cubic | |
|
|
| kornia.geometry.solvers.polynomial_solver.multiply_deg_one_poly | |
|
|
| kornia.geometry.solvers.polynomial_solver.multiply_deg_two_one_poly | |
|
|
| kornia.geometry.solvers.polynomial_solver.determinant_to_polynomial | |
|
|
geometry.subpix
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.subpix.spatial_soft_argmax.conv_soft_argmax2d | |
|
|
| kornia.geometry.subpix.spatial_soft_argmax.conv_soft_argmax3d | |
|
|
| kornia.geometry.subpix.spatial_soft_argmax.conv_quad_interp3d | |
|
|
| kornia.geometry.subpix.dsnt.spatial_softmax2d | |
|
|
| kornia.geometry.subpix.dsnt.spatial_expectation2d | |
|
|
| kornia.geometry.subpix.spatial_soft_argmax.spatial_soft_argmax2d | |
|
|
| kornia.geometry.subpix.dsnt.render_gaussian2d | |
|
|
| kornia.geometry.subpix.nms.nms2d | |
|
|
| kornia.geometry.subpix.nms.nms3d | |
|
|
| kornia.geometry.subpix.SpatialSoftArgmax2d | |
|
|
| kornia.geometry.subpix.ConvSoftArgmax2d | |
|
|
| kornia.geometry.subpix.ConvSoftArgmax3d | |
|
|
| kornia.geometry.subpix.ConvQuadInterp3d | |
|
|
| kornia.geometry.subpix.NonMaximaSuppression2d | |
|
|
| kornia.geometry.subpix.NonMaximaSuppression3d | |
|
|
geometry.transform
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.transform.imgwarp.warp_perspective | |
|
|
| kornia.geometry.transform.imgwarp.warp_perspective3d | |
|
|
| kornia.geometry.transform.imgwarp.warp_affine | |
|
|
| kornia.geometry.transform.imgwarp.warp_affine3d | |
|
|
| kornia.geometry.transform.thin_plate_spline.warp_image_tps | |
|
|
| kornia.geometry.transform.thin_plate_spline.warp_points_tps | |
|
|
| kornia.geometry.transform.imgwarp.warp_grid | |
|
|
| kornia.geometry.transform.imgwarp.warp_grid3d | |
|
|
| kornia.geometry.transform.imgwarp.remap | |
|
|
| kornia.geometry.transform.affwarp.affine | |
|
|
| kornia.geometry.transform.affwarp.rotate | |
|
|
| kornia.geometry.transform.affwarp.translate | |
|
|
| kornia.geometry.transform.affwarp.scale | |
|
|
| kornia.geometry.transform.affwarp.shear | |
|
|
| kornia.geometry.transform.flips.hflip | |
|
|
| kornia.geometry.transform.flips.vflip | |
|
|
| kornia.geometry.transform.flips.rot180 | |
|
|
| kornia.geometry.transform.affwarp.resize | |
|
|
| kornia.geometry.transform.affwarp.rescale | |
|
|
| kornia.geometry.transform.elastic_transform.elastic_transform2d | |
|
|
| kornia.geometry.transform.pyramid.pyrdown | |
|
|
| kornia.geometry.transform.pyramid.pyrup | |
|
|
| kornia.geometry.transform.pyramid.build_pyramid | |
|
|
| kornia.geometry.transform.pyramid.build_laplacian_pyramid | |
|
|
| kornia.geometry.transform.pyramid.upscale_double | |
|
|
| kornia.geometry.transform.imgwarp.get_perspective_transform | |
|
|
| kornia.geometry.transform.imgwarp.get_perspective_transform3d | |
|
|
| kornia.geometry.transform.imgwarp.get_projective_transform | |
|
|
| kornia.geometry.transform.imgwarp.get_rotation_matrix2d | |
|
|
| kornia.geometry.transform.imgwarp.get_shear_matrix2d | |
|
|
| kornia.geometry.transform.imgwarp.get_shear_matrix3d | |
|
|
| kornia.geometry.transform.imgwarp.get_affine_matrix2d | |
|
|
| kornia.geometry.transform.imgwarp.get_affine_matrix3d | |
|
|
| kornia.geometry.transform.imgwarp.invert_affine_transform | |
|
|
| kornia.geometry.transform.imgwarp.projection_from_Rt | |
|
|
| kornia.geometry.transform.thin_plate_spline.get_tps_transform | |
|
|
| kornia.geometry.transform.crop2d.crop_by_indices | |
|
|
| kornia.geometry.transform.crop2d.crop_by_boxes | |
|
|
| kornia.geometry.transform.crop2d.center_crop | |
|
|
| kornia.geometry.transform.crop2d.crop_and_resize | |
|
|
| kornia.geometry.transform.Rotate | |
|
|
| kornia.geometry.transform.Translate | |
|
|
| kornia.geometry.transform.Scale | |
|
|
| kornia.geometry.transform.Shear | |
|
|
| kornia.geometry.transform.PyrDown | |
|
|
| kornia.geometry.transform.PyrUp | |
|
|
| kornia.geometry.transform.ScalePyramid | |
|
|
| kornia.geometry.transform.Hflip | |
|
|
| kornia.geometry.transform.Vflip | |
|
|
| kornia.geometry.transform.Rot180 | |
|
|
| kornia.geometry.transform.Resize | |
|
|
| kornia.geometry.transform.Rescale | |
|
|
| kornia.geometry.transform.Affine | |
|
|
| kornia.geometry.transform.HomographyWarper | |
|
|
| kornia.geometry.transform.Homography | |
|
|
| kornia.geometry.transform.Similarity | |
|
|
geometry.vector
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.geometry.vector.Vector3 | |
|
|
| kornia.geometry.vector.Vector2 | |
|
|
image
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.image.ImageSize | |
|
|
| kornia.image.PixelFormat | |
|
|
| kornia.image.ChannelsOrder | |
|
|
| kornia.image.ImageLayout | |
|
|
| kornia.image.Image | |
|
|
| kornia.image.Image.from_numpy | |
|
|
| kornia.image.Image.from_dlpack | |
|
|
| kornia.image.Image.to_numpy | |
|
|
io
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.io.load_image | |
|
|
| kornia.io.write_image | |
|
|
losses
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.losses.ssim.ssim_loss | |
|
|
| kornia.losses.ssim3d.ssim3d_loss | |
|
|
| kornia.losses.psnr.psnr_loss | |
|
|
| kornia.losses.total_variation.total_variation | |
|
|
| kornia.losses.depth_smooth.inverse_depth_smoothness_loss | |
|
|
| kornia.losses.charbonnier.charbonnier_loss | |
|
|
| kornia.losses.welsch.welsch_loss | |
|
|
| kornia.losses.cauchy.cauchy_loss | |
|
|
| kornia.losses.geman_mcclure.geman_mcclure_loss | |
|
|
| kornia.losses.focal.binary_focal_loss_with_logits | |
|
|
| kornia.losses.focal.focal_loss | |
|
|
| kornia.losses.dice.dice_loss | |
|
|
| kornia.losses.tversky.tversky_loss | |
|
|
| kornia.losses.lovasz_hinge.lovasz_hinge_loss | |
|
|
| kornia.losses.lovasz_softmax.lovasz_softmax_loss | |
|
|
| kornia.losses.divergence.js_div_loss_2d | |
|
|
| kornia.losses.divergence.kl_div_loss_2d | |
|
|
| kornia.losses.HausdorffERLoss | |
|
|
| kornia.losses.HausdorffERLoss3D | |
|
|
| kornia.losses.SSIMLoss | |
|
|
| kornia.losses.SSIM3DLoss | |
|
|
| kornia.losses.MS_SSIMLoss | |
|
|
| kornia.losses.TotalVariation | |
|
|
| kornia.losses.PSNRLoss | |
|
|
| kornia.losses.InverseDepthSmoothnessLoss | |
|
|
| kornia.losses.CharbonnierLoss | |
|
|
| kornia.losses.WelschLoss | |
|
|
| kornia.losses.CauchyLoss | |
|
|
| kornia.losses.GemanMcclureLoss | |
|
|
| kornia.losses.BinaryFocalLossWithLogits | |
|
|
| kornia.losses.DiceLoss | |
|
|
| kornia.losses.TverskyLoss | |
|
|
| kornia.losses.FocalLoss | |
|
|
| kornia.losses.LovaszHingeLoss | |
|
|
| kornia.losses.LovaszSoftmaxLoss | |
|
|
metrics
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.metrics.accuracy.accuracy | |
|
|
| kornia.metrics.confusion_matrix.confusion_matrix | |
|
|
| kornia.metrics.mean_iou.mean_iou | |
|
|
| kornia.metrics.mean_average_precision.mean_average_precision | |
|
|
| kornia.metrics.mean_iou.mean_iou_bbox | |
|
|
| kornia.metrics.psnr.psnr | |
|
|
| kornia.metrics.ssim.ssim | |
|
|
| kornia.metrics.ssim3d.ssim3d | |
|
|
| kornia.metrics.endpoint_error.aepe | |
|
|
| kornia.metrics.SSIM | |
|
|
| kornia.metrics.SSIM3D | |
|
|
| kornia.metrics.AEPE | |
|
|
| kornia.metrics.AverageMeter | |
|
|
morphology
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.morphology.morphology.dilation | |
|
|
| kornia.morphology.morphology.erosion | |
|
|
| kornia.morphology.morphology.opening | |
|
|
| kornia.morphology.morphology.closing | |
|
|
| kornia.morphology.morphology.gradient | |
|
|
| kornia.morphology.morphology.top_hat | |
|
|
| kornia.morphology.morphology.bottom_hat | |
|
|
nerf
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.nerf.nerf_model.NerfModel | |
|
|
| kornia.nerf.nerf_model.NerfModelRenderer | |
|
|
| kornia.nerf.nerf_model.MLP | |
|
|
| kornia.nerf.volume_renderer.IrregularRenderer | |
|
|
| kornia.nerf.volume_renderer.RegularRenderer | |
|
|
| kornia.nerf.samplers.RandomRaySampler | |
|
|
sensors
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.sensors.camera.CameraModel | |
|
|
| kornia.sensors.camera.PinholeModel | |
|
|
| kornia.sensors.camera.distortion_model.AffineTransform | |
|
|
| kornia.sensors.camera.projection_model.Z1Projection | |
|
|
tracking
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.tracking.HomographyTracker | |
|
|
utils
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.utils.draw.draw_line | |
|
|
| kornia.utils.draw.draw_rectangle | |
|
|
| kornia.utils.draw.draw_convex_polygon | |
|
|
| kornia.utils.grid.create_meshgrid | |
|
|
| kornia.utils.grid.create_meshgrid3d | |
|
|
| kornia.utils.one_hot.one_hot | |
|
|
| kornia.utils.tensor_to_image | |
|
|
| kornia.utils.image_to_tensor | |
|
|
| kornia.utils.image_list_to_tensor | |
|
|
| kornia.utils.image_to_string | |
|
|
| kornia.utils.save_pointcloud_ply | |
|
|
| kornia.utils.map_location_to_cpu | |
|
|
| kornia.utils.is_autocast_enabled | |
|
|
x
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| kornia.x.ImageClassifierTrainer | |
|
|
| kornia.x.SemanticSegmentationTrainer | |
|
|
| kornia.x.ObjectDetectionTrainer | |
|
|
| kornia.x.ModelCheckpoint | |
|
|
| kornia.x.EarlyStopping | |
|
|
transformers
vision
| Function | numpy | jax | tensorflow |
|---|---|---|---|
| transformers.Swin2SRModel |  |
|
|