Spaces:
Running
Running
make examples more clear
Browse files
app.py
CHANGED
@@ -136,25 +136,23 @@ def inference(image1, image2, task='flow'):
|
|
136 |
|
137 |
title = "UniMatch"
|
138 |
|
139 |
-
description = "<p style='text-align: center'>Optical flow and stereo matching demo for <a href='https://haofeixu.github.io/unimatch/' target='_blank'>Unifying Flow, Stereo and Depth Estimation</a> | <a href='https://arxiv.org/abs/2211.05783' target='_blank'>Paper</a> | <a href='https://github.com/autonomousvision/unimatch' target='_blank'>Code</a> | <a href='https://colab.research.google.com/drive/1r5m-xVy3Kw60U-m5VB-aQ98oqqg_6cab?usp=sharing' target='_blank'>Colab</a><br>
|
140 |
|
141 |
examples = [
|
142 |
-
['demo/flow_kitti_test_000197_10.png', 'demo/flow_kitti_test_000197_11.png'],
|
143 |
-
['demo/flow_sintel_cave_3_frame_0049.png', 'demo/flow_sintel_cave_3_frame_0050.png'],
|
144 |
-
['demo/flow_davis_skate-jump_00059.jpg', 'demo/flow_davis_skate-jump_00060.jpg'],
|
145 |
['demo/stereo_drivingstereo_test_2018-07-11-14-48-52_2018-07-11-14-58-34-673_left.jpg',
|
146 |
-
'demo/stereo_drivingstereo_test_2018-07-11-14-48-52_2018-07-11-14-58-34-673_right.jpg'],
|
147 |
-
['demo/stereo_middlebury_plants_im0.png', 'demo/stereo_middlebury_plants_im1.png'],
|
148 |
-
['demo/stereo_holopix_left.png', 'demo/stereo_holopix_right.png']
|
|
|
|
|
|
|
149 |
]
|
150 |
|
151 |
gr.Interface(
|
152 |
inference,
|
153 |
[gr.Image(type="pil", label="Image1"), gr.Image(type="pil", label="Image2"), gr.Radio(choices=['flow', 'stereo'], value='flow', label='Task')],
|
154 |
-
gr.Image(type="pil", label="
|
155 |
title=title,
|
156 |
description=description,
|
157 |
examples=examples,
|
158 |
-
|
159 |
-
allow_flagging="auto",
|
160 |
-
).launch(debug=True)
|
|
|
136 |
|
137 |
title = "UniMatch"
|
138 |
|
139 |
+
description = "<p style='text-align: center'>Optical flow and stereo matching demo for <a href='https://haofeixu.github.io/unimatch/' target='_blank'>Unifying Flow, Stereo and Depth Estimation</a> | <a href='https://arxiv.org/abs/2211.05783' target='_blank'>Paper</a> | <a href='https://github.com/autonomousvision/unimatch' target='_blank'>Code</a> | <a href='https://colab.research.google.com/drive/1r5m-xVy3Kw60U-m5VB-aQ98oqqg_6cab?usp=sharing' target='_blank'>Colab</a><br>Task <strong>flow</strong>: Image1: <strong>video frame t</strong>, Image2: <strong>video frame t+1</strong>; Task <strong>stereo</strong>: Image1: <strong>left</strong> image, Image2: <strong>right</strong> image<br>Simply upload your images or click one of the provided examples.<br><strong>Select the task type according to your input images</strong>.</p>"
|
140 |
|
141 |
examples = [
|
|
|
|
|
|
|
142 |
['demo/stereo_drivingstereo_test_2018-07-11-14-48-52_2018-07-11-14-58-34-673_left.jpg',
|
143 |
+
'demo/stereo_drivingstereo_test_2018-07-11-14-48-52_2018-07-11-14-58-34-673_right.jpg', 'stereo'],
|
144 |
+
['demo/stereo_middlebury_plants_im0.png', 'demo/stereo_middlebury_plants_im1.png', 'stereo'],
|
145 |
+
['demo/stereo_holopix_left.png', 'demo/stereo_holopix_right.png', 'stereo'],
|
146 |
+
['demo/flow_kitti_test_000197_10.png', 'demo/flow_kitti_test_000197_11.png', 'flow'],
|
147 |
+
['demo/flow_sintel_cave_3_frame_0049.png', 'demo/flow_sintel_cave_3_frame_0050.png', 'flow'],
|
148 |
+
['demo/flow_davis_skate-jump_00059.jpg', 'demo/flow_davis_skate-jump_00060.jpg', 'flow']
|
149 |
]
|
150 |
|
151 |
gr.Interface(
|
152 |
inference,
|
153 |
[gr.Image(type="pil", label="Image1"), gr.Image(type="pil", label="Image2"), gr.Radio(choices=['flow', 'stereo'], value='flow', label='Task')],
|
154 |
+
gr.Image(type="pil", label="flow/disparity"),
|
155 |
title=title,
|
156 |
description=description,
|
157 |
examples=examples,
|
158 |
+
).launch(debug=True, quiet=True)
|
|
|
|