mattmdjaga
commited on
Commit
·
87e8319
1
Parent(s):
95b7681
Added license and citation
Browse files
README.md
CHANGED
@@ -43,4 +43,31 @@ upsampled_logits = nn.functional.interpolate(
|
|
43 |
|
44 |
pred_seg = upsampled_logits.argmax(dim=1)[0]
|
45 |
plt.imshow(pred_seg)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
46 |
```
|
|
|
43 |
|
44 |
pred_seg = upsampled_logits.argmax(dim=1)[0]
|
45 |
plt.imshow(pred_seg)
|
46 |
+
```
|
47 |
+
### License
|
48 |
+
|
49 |
+
The license for this model can be found [here](https://github.com/NVlabs/SegFormer/blob/master/LICENSE).
|
50 |
+
|
51 |
+
### BibTeX entry and citation info
|
52 |
+
|
53 |
+
```bibtex
|
54 |
+
@article{DBLP:journals/corr/abs-2105-15203,
|
55 |
+
author = {Enze Xie and
|
56 |
+
Wenhai Wang and
|
57 |
+
Zhiding Yu and
|
58 |
+
Anima Anandkumar and
|
59 |
+
Jose M. Alvarez and
|
60 |
+
Ping Luo},
|
61 |
+
title = {SegFormer: Simple and Efficient Design for Semantic Segmentation with
|
62 |
+
Transformers},
|
63 |
+
journal = {CoRR},
|
64 |
+
volume = {abs/2105.15203},
|
65 |
+
year = {2021},
|
66 |
+
url = {https://arxiv.org/abs/2105.15203},
|
67 |
+
eprinttype = {arXiv},
|
68 |
+
eprint = {2105.15203},
|
69 |
+
timestamp = {Wed, 02 Jun 2021 11:46:42 +0200},
|
70 |
+
biburl = {https://dblp.org/rec/journals/corr/abs-2105-15203.bib},
|
71 |
+
bibsource = {dblp computer science bibliography, https://dblp.org}
|
72 |
+
}
|
73 |
```
|