Spaces:
Sleeping
Sleeping
Realcat
commited on
Commit
•
7419d98
1
Parent(s):
8d7004c
update: config
Browse files- common/app_class.py +8 -1
- 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=
|
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:
|