Update asap.py

fix issue in im_feeling_lucky
AndrewAnnex · Jul 2, 2021 · 5f4f8c5 · 5f4f8c5
1 parent 53a026a
commit 5f4f8c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/asap_stereo/asap.py b/asap_stereo/asap.py
@@ -1762,9 +1762,9 @@ def im_feeling_lucky(self, ref_img, mobile_image, *other_mobile, ipfindkwargs=No
         If unsure normalize your data ahead of time
         """
         # get the matches
-        self.find_matches(ref_img, mobile_image, ipfindkwargs=ipfindkwargs, ipmatchkwargs=ipmatchkwargs)
+        matches = list(self.find_matches(ref_img, mobile_image, ipfindkwargs=ipfindkwargs, ipmatchkwargs=ipmatchkwargs))
         # convert matches to csv
-        match_csv = self.matches_to_csv(f'{Path(ref_img).stem}__{Path(mobile_image).stem}.match')
+        match_csv = self.matches_to_csv(matches[0])
         # loop through all the mobile data
         for i, mobile in tqdm.tqdm(enumerate([mobile_image, *other_mobile])):
             # transform matches # todo: make sure I don't overwrite anything here