ZhengPeng7 commited on
Commit
398e876
·
1 Parent(s): 726f866

Polish up the interface. Correct the specified examples.

Browse files
Files changed (2) hide show
  1. app.py +4 -1
  2. 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