Realcat commited on
Commit
d64a873
1 Parent(s): 1e360c3

update: requirements

Browse files
Files changed (3) hide show
  1. app.py +3 -6
  2. pre-requirements.txt +8 -8
  3. 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 image
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=0.99999,
214
  )
215
  ransac_max_iter = gr.Slider(
216
  minimum=0.0,
217
  maximum=100000,
218
  step=100,
219
  label="Ransac Iterations",
220
- value=10000,
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>=1.12.1
2
- torchvision>=0.13.1
3
- torchmetrics>=0.6.0
4
- opencv-python>=4.5.5.64
5
- pytorch-lightning>=1.4.2
6
- omegaconf>=2.1.1
7
- einops>=0.3.0
8
- kornia>=0.6
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>=0.5.0
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