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'],
|