ensuring gray conversion works with kornia 0.6.3

assignment_8_9_tracking_template/klt.py

@@ -13,7 +13,7 @@ import matplotlib.pyplot as plt
 def read_image(idx): 
     fname = os.path.join('data', '%03i.jpg' % idx)
     img = cv2.imread(fname)
-    img = kornia.color.bgr_to_grayscale(kornia.image_to_tensor(img,False))/255.0
+    img = kornia.color.bgr_to_grayscale(kornia.image_to_tensor(img.astype(np.float32),False))/255.0
     return img
 
 def get_patches_subpix(img: torch.Tensor, 

assignment_8_9_tracking_template/test.py

@@ -17,7 +17,7 @@ translations = [ [0, 0],
 def get_test_data(idx): 
     fname = os.path.join('test_data', 'toy_im_%01i.png' % idx)
     img = cv2.imread(fname)
-    img = kornia.color.bgr_to_grayscale(kornia.image_to_tensor(img,False))/255.0
+    img = kornia.color.bgr_to_grayscale(kornia.image_to_tensor(img.astype(np.float32),False))/255.0
     return img, translations[idx]