gent
commited on
Commit
·
f38ae12
1
Parent(s):
ed01e82
Update README.md with SiT model information
Browse files
README.md
CHANGED
@@ -1,3 +1,43 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
tags:
|
4 |
+
- self-supervised learning
|
5 |
+
- vision
|
6 |
+
- SiT
|
7 |
+
inference: false
|
8 |
+
|
9 |
---
|
10 |
+
|
11 |
+
# Model description
|
12 |
+
|
13 |
+
SiT is a self-supervised learning model that combines masked image modeling and contrastive learning. The model is trained on ImageNet-1K.
|
14 |
+
|
15 |
+
|
16 |
+
# Model Sources
|
17 |
+
|
18 |
+
- https://github.com/Sara-Ahmed/SiT
|
19 |
+
- https://arxiv.org/abs/2104.03602
|
20 |
+
|
21 |
+
|
22 |
+
# Model Card Authors
|
23 |
+
Sara Atito, Muhammad Awais, Josef Kittler
|
24 |
+
|
25 |
+
# How to use
|
26 |
+
|
27 |
+
```python
|
28 |
+
from modeling_sit import ViTSiTForPreTraining
|
29 |
+
# reload
|
30 |
+
model = ViTSiTForPreTraining.from_pretrained("erow/SiT")
|
31 |
+
```
|
32 |
+
|
33 |
+
# BibTeX entry and citation info
|
34 |
+
```
|
35 |
+
@inproceedings{atito2023sit,
|
36 |
+
title={SiT is all you need},
|
37 |
+
author={Atito, Sara and Awais, Muhammed and Nandam, Srinivasa and Kittler, Josef},
|
38 |
+
booktitle={2023 IEEE International Conference on Image Processing (ICIP)},
|
39 |
+
pages={2125--2129},
|
40 |
+
year={2023},
|
41 |
+
organization={IEEE}
|
42 |
+
}
|
43 |
+
```
|