vobecant
commited on
Commit
·
350d469
1
Parent(s):
62f24a8
Initial commit.
Browse files- app.py +1 -1
- weights/segmenter_nusc.pth_variant.yml +1 -1
app.py
CHANGED
|
@@ -159,7 +159,7 @@ def predict(input_img):
|
|
| 159 |
title = "Drive&Segment"
|
| 160 |
description = 'Gradio Demo accompanying paper "Drive&Segment: Unsupervised Semantic Segmentation of Urban Scenes via Cross-modal Distillation"\nBecause of the CPU-only inference, it might take up to 20s for large images.'
|
| 161 |
article = "<p style='text-align: center'><a href='TODO' target='_blank'>Project Page</a> | <a href='codelink' target='_blank'>Github</a></p>"
|
| 162 |
-
examples = ['examples/img1.jpg'
|
| 163 |
|
| 164 |
# predict(examples[0])
|
| 165 |
|
|
|
|
| 159 |
title = "Drive&Segment"
|
| 160 |
description = 'Gradio Demo accompanying paper "Drive&Segment: Unsupervised Semantic Segmentation of Urban Scenes via Cross-modal Distillation"\nBecause of the CPU-only inference, it might take up to 20s for large images.'
|
| 161 |
article = "<p style='text-align: center'><a href='TODO' target='_blank'>Project Page</a> | <a href='codelink' target='_blank'>Github</a></p>"
|
| 162 |
+
examples = ['examples/img1.jpg'] #, 'examples/img2.jpeg']
|
| 163 |
|
| 164 |
# predict(examples[0])
|
| 165 |
|
weights/segmenter_nusc.pth_variant.yml
CHANGED
|
@@ -5,7 +5,7 @@ debug: false
|
|
| 5 |
inference_kwargs:
|
| 6 |
im_size: 1024
|
| 7 |
window_size: 512
|
| 8 |
-
window_stride:
|
| 9 |
net_kwargs:
|
| 10 |
backbone: dino_vits16
|
| 11 |
d_model: 384
|
|
|
|
| 5 |
inference_kwargs:
|
| 6 |
im_size: 1024
|
| 7 |
window_size: 512
|
| 8 |
+
window_stride: 384
|
| 9 |
net_kwargs:
|
| 10 |
backbone: dino_vits16
|
| 11 |
d_model: 384
|