EduardoPacheco
commited on
Commit
•
3384168
1
Parent(s):
05a5d2c
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
---
|
2 |
-
license: cc-by-nc-4.0
|
3 |
-
library_name: transformers
|
4 |
datasets:
|
5 |
- imagenet-1k
|
6 |
-
pipeline_tag: image-classification
|
7 |
language:
|
8 |
- en
|
|
|
|
|
|
|
9 |
---
|
10 |
# Hiera Model (Tiny, fine-tuned on IN1K)
|
11 |
|
@@ -27,7 +27,7 @@ We show that a lot of this bulk is actually _unnecessary_. Instead of manually a
|
|
27 |
|
28 |
## Intended uses & limitations
|
29 |
|
30 |
-
Hiera can be used for image classification, feature extraction or masked image modeling. This checkpoint in specific is intended for **Image Classificaiton
|
31 |
|
32 |
### How to use
|
33 |
|
@@ -55,6 +55,7 @@ predicted_id = outputs.logits.argmax(dim=-1).item()
|
|
55 |
predicted_class = model.config.id2label[predicted_id] # 'tabby, tabby cat'
|
56 |
```
|
57 |
|
|
|
58 |
### BibTeX entry and citation info
|
59 |
If you use Hiera or this code in your work, please cite:
|
60 |
```
|
@@ -63,5 +64,4 @@ If you use Hiera or this code in your work, please cite:
|
|
63 |
author={Ryali, Chaitanya and Hu, Yuan-Ting and Bolya, Daniel and Wei, Chen and Fan, Haoqi and Huang, Po-Yao and Aggarwal, Vaibhav and Chowdhury, Arkabandhu and Poursaeed, Omid and Hoffman, Judy and Malik, Jitendra and Li, Yanghao and Feichtenhofer, Christoph},
|
64 |
journal={ICML},
|
65 |
year={2023}
|
66 |
-
}
|
67 |
-
```
|
|
|
1 |
---
|
|
|
|
|
2 |
datasets:
|
3 |
- imagenet-1k
|
|
|
4 |
language:
|
5 |
- en
|
6 |
+
library_name: transformers
|
7 |
+
license: cc-by-nc-4.0
|
8 |
+
pipeline_tag: image-classification
|
9 |
---
|
10 |
# Hiera Model (Tiny, fine-tuned on IN1K)
|
11 |
|
|
|
27 |
|
28 |
## Intended uses & limitations
|
29 |
|
30 |
+
Hiera can be used for image classification, feature extraction or masked image modeling. This checkpoint in specific is intended for **Image Classificaiton**.
|
31 |
|
32 |
### How to use
|
33 |
|
|
|
55 |
predicted_class = model.config.id2label[predicted_id] # 'tabby, tabby cat'
|
56 |
```
|
57 |
|
58 |
+
|
59 |
### BibTeX entry and citation info
|
60 |
If you use Hiera or this code in your work, please cite:
|
61 |
```
|
|
|
64 |
author={Ryali, Chaitanya and Hu, Yuan-Ting and Bolya, Daniel and Wei, Chen and Fan, Haoqi and Huang, Po-Yao and Aggarwal, Vaibhav and Chowdhury, Arkabandhu and Poursaeed, Omid and Hoffman, Judy and Malik, Jitendra and Li, Yanghao and Feichtenhofer, Christoph},
|
65 |
journal={ICML},
|
66 |
year={2023}
|
67 |
+
}
|
|