lsxi77777 commited on
Commit
ad0670f
·
1 Parent(s): 0f1f833
Files changed (3) hide show
  1. output.pkl +2 -2
  2. ui/app_class.py +1 -1
  3. ui/utils.py +2 -0
output.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:faf509c60b0b906ac4e6837b1632a9567d602f04ee069cddc384d617e8d623e5
3
- size 2792855
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb391febee7e6824ce10523a189a190a7a6d739a56dabbf9ad54d088ade2e608
3
+ size 2718884
ui/app_class.py CHANGED
@@ -126,7 +126,7 @@ class ImageMatchingApp:
126
  with gr.Row():
127
  matcher_list = gr.Dropdown(
128
  choices=self.init_matcher_dropdown(),
129
- value="superpoint-minima_lightglue",
130
  label="Matching Model",
131
  interactive=True,
132
  )
 
126
  with gr.Row():
127
  matcher_list = gr.Dropdown(
128
  choices=self.init_matcher_dropdown(),
129
+ value="superpoint+minima(lightglue)",
130
  label="Matching Model",
131
  interactive=True,
132
  )
ui/utils.py CHANGED
@@ -594,6 +594,7 @@ def wrap_images(
594
  rectified_image1 = cv2.warpPerspective(img1, H, (w0, h0))
595
  title = ["Image 0", "Image 1 - warped"]
596
  elif geom_type == "Fundamental":
 
597
  if geom_type not in geo_info:
598
  logger.warning(f"{geom_type} not exist, maybe too less matches")
599
  return None, None
@@ -602,6 +603,7 @@ def wrap_images(
602
  rectified_image0 = cv2.warpPerspective(img0, H1, (w0, h0))
603
  rectified_image1 = cv2.warpPerspective(img1, H2, (w1, h1))
604
  title = ["Image 0 - warped", "Image 1 - warped"]
 
605
  else:
606
  print("Error: Unknown geometry type")
607
  fig = plot_images(
 
594
  rectified_image1 = cv2.warpPerspective(img1, H, (w0, h0))
595
  title = ["Image 0", "Image 1 - warped"]
596
  elif geom_type == "Fundamental":
597
+ print('if geom_type not in geo_info', geom_type not in geo_info)
598
  if geom_type not in geo_info:
599
  logger.warning(f"{geom_type} not exist, maybe too less matches")
600
  return None, None
 
603
  rectified_image0 = cv2.warpPerspective(img0, H1, (w0, h0))
604
  rectified_image1 = cv2.warpPerspective(img1, H2, (w1, h1))
605
  title = ["Image 0 - warped", "Image 1 - warped"]
606
+ print('success')
607
  else:
608
  print("Error: Unknown geometry type")
609
  fig = plot_images(