Spaces:
Running
on
Zero
Running
on
Zero
ZhengPeng7
commited on
Commit
·
398e876
1
Parent(s):
726f866
Polish up the interface. Correct the specified examples.
Browse files- app.py +4 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -64,7 +64,7 @@ def pred_maps(dr):
|
|
64 |
return image_preds[:]
|
65 |
|
66 |
|
67 |
-
examples = glob('example_images/butterfly')
|
68 |
|
69 |
ipt = ['image' for _ in range(5)]
|
70 |
opt = ipt.copy()
|
@@ -73,5 +73,8 @@ demo = gr.Interface(
|
|
73 |
inputs=ipt,
|
74 |
outputs=opt,
|
75 |
examples=examples,
|
|
|
|
|
|
|
76 |
)
|
77 |
demo.launch(debug=True)
|
|
|
64 |
return image_preds[:]
|
65 |
|
66 |
|
67 |
+
examples = glob('example_images/butterfly/*')
|
68 |
|
69 |
ipt = ['image' for _ in range(5)]
|
70 |
opt = ipt.copy()
|
|
|
73 |
inputs=ipt,
|
74 |
outputs=opt,
|
75 |
examples=examples,
|
76 |
+
interpretation='default',
|
77 |
+
title='Online demo for `GCoNet+: A Stronger Group Collaborative Co-Salient Object Detector (T-PAMI 2023)`',
|
78 |
+
description='Upload pictures, most of which contain salient objects of the same class.\nOur demo will give you the binary maps of these co-salient objects :)'
|
79 |
)
|
80 |
demo.launch(debug=True)
|
requirements.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
torch==1.13.1
|
2 |
torchvision==0.14.1
|
3 |
opencv-python
|
4 |
-
fvcore
|
|
|
|
1 |
torch==1.13.1
|
2 |
torchvision==0.14.1
|
3 |
opencv-python
|
4 |
+
fvcore
|
5 |
+
scikit-image
|