use kornia.io to load image and preprocess (#2)

- use kornia.io to load image and preprocess (4f612641d683f1a7919689893da48cc33b3c7cd4)


Co-authored-by: Edgar Riba <edgarriba@users.noreply.huggingface.co>

app.py

@@ -8,20 +8,17 @@ from kornia_moons.feature import *
 import gradio as gr
 
 def load_torch_image(fname):
-    img = K.image_to_tensor(fname, False).float() /255.
-    img = K.color.bgr_to_rgb(img)
+    img: Tensor = K.io.load_image(fname, K.io.ImageLoadType.RGB32)
+    img = img[None]  # 1xCxHxW / fp32 / [0, 1]
+    img = K.geometry.resize(img, (700, 700))
     return img
+ 
 def inference(file1,file2):
     fname1 = file1.name
-    fname2 = file2.name
-    im1 = cv2.imread(fname1)                 
-    im1re = cv2.resize(im1, (700, 700))  
-    im2 = cv2.imread(fname2)                 
-    im2re = cv2.resize(im2, (700, 700))       
+    fname2 = file2.name   
     img1 = load_torch_image(im1re)
     img2 = load_torch_image(im2re)
 
-
     matcher = KF.LoFTR(pretrained='outdoor')
 
     input_dict = {"image0": K.color.rgb_to_grayscale(img1), # LofTR works on grayscale images only