Working on ImgMch, 3

amydaali · Jul 4, 2019 · 69e8373 · 69e8373
1 parent c79ac1a
commit 69e8373
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/gluon/datasets/hpatches_mch_dataset.py b/gluon/datasets/hpatches_mch_dataset.py
@@ -59,8 +59,10 @@ def __init__(self,
     def __getitem__(self, index):
         # image = cv2.imread(self.image_paths[index], flags=cv2.IMREAD_GRAYSCALE)
         # warped_image = cv2.imread(self.warped_image_paths[index], flags=cv2.IMREAD_GRAYSCALE)
-        image = mx.image.imread(self.image_paths[index], flag=0)
-        warped_image = mx.image.imread(self.warped_image_paths[index], flag=0)
+        # image = mx.image.imread(self.image_paths[index], flag=0)
+        # warped_image = mx.image.imread(self.warped_image_paths[index], flag=0)
+        image = mx.nd.array(cv2.imread(self.image_paths[index], flags=0))
+        warped_image = mx.nd.array(cv2.imread(self.warped_image_paths[index], flags=0))
         homography = mx.nd.array(self.homographies[index])
 
         if self.transform is not None: