Skip to content

Conversions between kornia and other computer vision libraries formats

License

Notifications You must be signed in to change notification settings

ducha-aiki/kornia_moons

Repository files navigation

Kornia_moons

Install

pip install kornia_moons

How to use

Here is an example how to use kornia-moons for local feature conversion from OpenCV ORB keypoints

import matplotlib.pyplot as plt
import cv2
import torch
import kornia as K
from typing import List
import matplotlib.pyplot as plt

from kornia_moons.feature import laf_from_opencv_ORB_kpts, opencv_ORB_kpts_from_laf 
from kornia_moons.viz import visualize_LAF



img = cv2.cvtColor(cv2.imread('data/strahov.png'), cv2.COLOR_BGR2RGB)

det = cv2.ORB_create(500)
kps, descs = det.detectAndCompute(img, None)

out_img = cv2.drawKeypoints(img, kps, None, flags=cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS)
plt.imshow(out_img)


lafs = laf_from_opencv_ORB_kpts(kps)
visualize_LAF(K.image_to_tensor(img, False), lafs, 0)

kps_back = opencv_ORB_kpts_from_laf(lafs)
out_img2 = cv2.drawKeypoints(img, kps_back, None, flags=cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS)
plt.imshow(out_img2)
libpng warning: iCCP: known incorrect sRGB profile

<matplotlib.image.AxesImage>

About

Conversions between kornia and other computer vision libraries formats

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages