Spaces:
Running
Running
Realcat
commited on
Commit
•
d64a873
1
Parent(s):
1e360c3
update: requirements
Browse files- app.py +3 -6
- pre-requirements.txt +8 -8
- requirements.txt +16 -16
app.py
CHANGED
@@ -79,7 +79,7 @@ def ui_reset_state(*args):
|
|
79 |
None, # ransac matches
|
80 |
{}, # matches result info
|
81 |
{}, # matcher config
|
82 |
-
None, # warped
|
83 |
{}, # geometry result
|
84 |
DEFAULT_RANSAC_METHOD, # ransac_method
|
85 |
DEFAULT_RANSAC_REPROJ_THRESHOLD, # ransac_reproj_threshold
|
@@ -191,7 +191,6 @@ def run(server_name="0.0.0.0", server_port=7860):
|
|
191 |
# )
|
192 |
with gr.Accordion("RANSAC Setting", open=True):
|
193 |
with gr.Row(equal_height=False):
|
194 |
-
# enable_ransac = gr.Checkbox(label="Enable RANSAC")
|
195 |
ransac_method = gr.Dropdown(
|
196 |
choices=ransac_zoo.keys(),
|
197 |
value=DEFAULT_RANSAC_METHOD,
|
@@ -210,26 +209,24 @@ def run(server_name="0.0.0.0", server_port=7860):
|
|
210 |
maximum=1,
|
211 |
step=0.00001,
|
212 |
label="Ransac Confidence",
|
213 |
-
value=
|
214 |
)
|
215 |
ransac_max_iter = gr.Slider(
|
216 |
minimum=0.0,
|
217 |
maximum=100000,
|
218 |
step=100,
|
219 |
label="Ransac Iterations",
|
220 |
-
value=
|
221 |
)
|
222 |
|
223 |
with gr.Accordion("Geometry Setting", open=False):
|
224 |
with gr.Row(equal_height=False):
|
225 |
-
# show_geom = gr.Checkbox(label="Show Geometry")
|
226 |
choice_estimate_geom = gr.Radio(
|
227 |
["Fundamental", "Homography"],
|
228 |
label="Reconstruct Geometry",
|
229 |
value=DEFAULT_SETTING_GEOMETRY,
|
230 |
)
|
231 |
|
232 |
-
# with gr.Column():
|
233 |
# collect inputs
|
234 |
inputs = [
|
235 |
input_image0,
|
|
|
79 |
None, # ransac matches
|
80 |
{}, # matches result info
|
81 |
{}, # matcher config
|
82 |
+
None, # warped imageInstance of 'Radio' has no 'change' member
|
83 |
{}, # geometry result
|
84 |
DEFAULT_RANSAC_METHOD, # ransac_method
|
85 |
DEFAULT_RANSAC_REPROJ_THRESHOLD, # ransac_reproj_threshold
|
|
|
191 |
# )
|
192 |
with gr.Accordion("RANSAC Setting", open=True):
|
193 |
with gr.Row(equal_height=False):
|
|
|
194 |
ransac_method = gr.Dropdown(
|
195 |
choices=ransac_zoo.keys(),
|
196 |
value=DEFAULT_RANSAC_METHOD,
|
|
|
209 |
maximum=1,
|
210 |
step=0.00001,
|
211 |
label="Ransac Confidence",
|
212 |
+
value=DEFAULT_RANSAC_CONFIDENCE,
|
213 |
)
|
214 |
ransac_max_iter = gr.Slider(
|
215 |
minimum=0.0,
|
216 |
maximum=100000,
|
217 |
step=100,
|
218 |
label="Ransac Iterations",
|
219 |
+
value=DEFAULT_RANSAC_MAX_ITER,
|
220 |
)
|
221 |
|
222 |
with gr.Accordion("Geometry Setting", open=False):
|
223 |
with gr.Row(equal_height=False):
|
|
|
224 |
choice_estimate_geom = gr.Radio(
|
225 |
["Fundamental", "Homography"],
|
226 |
label="Reconstruct Geometry",
|
227 |
value=DEFAULT_SETTING_GEOMETRY,
|
228 |
)
|
229 |
|
|
|
230 |
# collect inputs
|
231 |
inputs = [
|
232 |
input_image0,
|
pre-requirements.txt
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
-
torch
|
2 |
-
torchvision
|
3 |
-
torchmetrics
|
4 |
-
opencv-python
|
5 |
-
pytorch-lightning
|
6 |
-
omegaconf
|
7 |
-
einops
|
8 |
-
kornia
|
9 |
gradio==3.41.2
|
10 |
gradio_client==0.5.0
|
|
|
1 |
+
torch==2.1.0
|
2 |
+
torchvision==0.16.0
|
3 |
+
torchmetrics==0.6.0
|
4 |
+
opencv-python==4.6.0.66
|
5 |
+
pytorch-lightning==1.4.9
|
6 |
+
omegaconf==2.3.0
|
7 |
+
einops==0.6.1
|
8 |
+
kornia==0.6.12
|
9 |
gradio==3.41.2
|
10 |
gradio_client==0.5.0
|
requirements.txt
CHANGED
@@ -1,17 +1,17 @@
|
|
1 |
numpy==1.23.5
|
2 |
-
tqdm
|
3 |
-
matplotlib
|
4 |
-
plotly
|
5 |
-
scipy
|
6 |
-
scikit_learn
|
7 |
-
scikit-image
|
8 |
-
h5py
|
9 |
-
pycolmap
|
10 |
-
gdown
|
11 |
-
seaborn
|
12 |
-
pytlsd
|
13 |
-
tensorboardX
|
14 |
-
shapely
|
15 |
-
yacs
|
16 |
-
loguru
|
17 |
-
e2cnn
|
|
|
1 |
numpy==1.23.5
|
2 |
+
tqdm==4.65.0
|
3 |
+
matplotlib==3.7.1
|
4 |
+
plotly==5.15.0
|
5 |
+
scipy==1.11.1
|
6 |
+
scikit_learn==1.2.2
|
7 |
+
scikit-image==0.21.0
|
8 |
+
h5py==3.9.0
|
9 |
+
pycolmap==0.5.0
|
10 |
+
gdown==4.7.1
|
11 |
+
seaborn==0.12.2
|
12 |
+
pytlsd==0.0.2
|
13 |
+
tensorboardX==2.6.1
|
14 |
+
shapely==2.0.1
|
15 |
+
yacs==0.1.8
|
16 |
+
loguru==0.7.0
|
17 |
+
e2cnn==0.2.3
|