wondervictor
commited on
Commit
·
4c769ac
1
Parent(s):
80fef41
update readme
Browse files
app.py
CHANGED
@@ -55,8 +55,17 @@ def pred(image_np, prompt):
|
|
55 |
|
56 |
return visualization / 255.0, pred_mask.astype(np.float16)
|
57 |
|
58 |
-
|
59 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
|
61 |
demo = gr.Interface(
|
62 |
fn=pred,
|
@@ -78,9 +87,8 @@ demo = gr.Interface(
|
|
78 |
"assets/carrots.jpg",
|
79 |
"3carrots in center with ice and greenn leaves"
|
80 |
]],
|
81 |
-
title="EVF-SAM: Referring Expression Segmentation",
|
82 |
-
description=
|
83 |
-
cache_examples=False,
|
84 |
allow_flagging="never")
|
85 |
# demo.launch()
|
86 |
demo.launch()
|
|
|
55 |
|
56 |
return visualization / 255.0, pred_mask.astype(np.float16)
|
57 |
|
58 |
+
|
59 |
+
desc = """
|
60 |
+
<div><h3>EVF-SAM: Early Vision-Language Fusion for Text-Prompted Segment Anything Model</h3>
|
61 |
+
<p>EVF-SAM extends SAM's capabilities with text-prompted segmentation, achieving high accuracy in Referring Expression Segmentation.</p></div>
|
62 |
+
<div style='display:flex; gap: 0.25rem; align-items: center'><a href="https://arxiv.org/abs/2406.20076"><img src="https://img.shields.io/badge/arXiv-Paper-red"></a><a href="https://github.com/hustvl/EVF-SAM"><img src="https://img.shields.io/badge/GitHub-Code-blue"></a></div>
|
63 |
+
"""
|
64 |
+
|
65 |
+
|
66 |
+
|
67 |
+
# desc_title_str = '<div align ="center"><img src="assets/logo.jpg" width="20%"><h3> Early Vision-Language Fusion for Text-Prompted Segment Anything Model</h3></div>'
|
68 |
+
# desc_link_str = '[![arxiv paper](https://img.shields.io/badge/arXiv-Paper-red)](https://arxiv.org/abs/2406.20076)'
|
69 |
|
70 |
demo = gr.Interface(
|
71 |
fn=pred,
|
|
|
87 |
"assets/carrots.jpg",
|
88 |
"3carrots in center with ice and greenn leaves"
|
89 |
]],
|
90 |
+
title="📷 EVF-SAM: Referring Expression Segmentation",
|
91 |
+
description=desc,
|
|
|
92 |
allow_flagging="never")
|
93 |
# demo.launch()
|
94 |
demo.launch()
|