Realcat commited on
Commit
7419d98
1 Parent(s): 8d7004c

update: config

Browse files
Files changed (2) hide show
  1. common/app_class.py +8 -1
  2. common/config.yaml +1 -0
common/app_class.py CHANGED
@@ -40,6 +40,13 @@ class ImageMatchingApp:
40
  self.init_interface()
41
  # print all the keys
42
 
 
 
 
 
 
 
 
43
  def init_interface(self):
44
  with gr.Blocks() as self.app:
45
  with gr.Row():
@@ -57,7 +64,7 @@ class ImageMatchingApp:
57
  with gr.Column():
58
  with gr.Row():
59
  matcher_list = gr.Dropdown(
60
- choices=list(self.matcher_zoo.keys()),
61
  value="disk+lightglue",
62
  label="Matching Model",
63
  interactive=True,
 
40
  self.init_interface()
41
  # print all the keys
42
 
43
+ def init_matcher_dropdown(self):
44
+ algos = []
45
+ for k, v in self.cfg["matcher_zoo"].items():
46
+ if v.get("enable", True):
47
+ algos.append(k)
48
+ return algos
49
+
50
  def init_interface(self):
51
  with gr.Blocks() as self.app:
52
  with gr.Row():
 
64
  with gr.Column():
65
  with gr.Row():
66
  matcher_list = gr.Dropdown(
67
+ choices=self.init_matcher_dropdown(),
68
  value="disk+lightglue",
69
  label="Matching Model",
70
  interactive=True,
common/config.yaml CHANGED
@@ -47,6 +47,7 @@ matcher_zoo:
47
  project: https://zju3dv.github.io/loftr
48
  display: true
49
  cotr:
 
50
  matcher: cotr
51
  dense: true
52
  info:
 
47
  project: https://zju3dv.github.io/loftr
48
  display: true
49
  cotr:
50
+ enable: false
51
  matcher: cotr
52
  dense: true
53
  info: