Nan Xue commited on
Commit
6715492
·
1 Parent(s): 821a6d0
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -53,6 +53,7 @@ def process_image(
53
  seed=0,
54
  randomize_seed=False
55
  ):
 
56
  from scalelsd.ssl.models.detector import ScaleLSD
57
  from scalelsd.base import show, WireframeGraph
58
  from scalelsd.ssl.misc.train_utils import fix_seeds, load_scalelsd_model
@@ -183,7 +184,7 @@ def run_demo():
183
 
184
  with gr.Column():
185
  draw_junctions_only = gr.Checkbox(False, label="Show Junctions Only")
186
- use_lsd = False #gr.Checkbox(False, label="Use LSD-Rectifier")
187
  use_nms = gr.Checkbox(True, label="Use NMS")
188
  output_format = gr.Dropdown(
189
  ['png', 'jpg', 'pdf'],
 
53
  seed=0,
54
  randomize_seed=False
55
  ):
56
+ use_lsd = False
57
  from scalelsd.ssl.models.detector import ScaleLSD
58
  from scalelsd.base import show, WireframeGraph
59
  from scalelsd.ssl.misc.train_utils import fix_seeds, load_scalelsd_model
 
184
 
185
  with gr.Column():
186
  draw_junctions_only = gr.Checkbox(False, label="Show Junctions Only")
187
+ use_lsd = gr.Checkbox(False, label="Use LSD-Rectifier")
188
  use_nms = gr.Checkbox(True, label="Use NMS")
189
  output_format = gr.Dropdown(
190
  ['png', 'jpg', 'pdf'],