Spaces:
Running
on
Zero
Running
on
Zero
Nan Xue
commited on
Commit
·
6715492
1
Parent(s):
821a6d0
update
Browse files
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 =
|
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'],
|