zuppif commited on
Commit
65cbec0
1 Parent(s): 69b0ec9

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -0
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - vision
5
+
6
+
7
+
8
+
9
+
10
+ ---
11
+
12
+ # RegNetModel
13
+
14
+ RegNetModel model was introduced in the paper [Self-supervised Pretraining of Visual Features in the Wild](https://arxiv.org/abs/2103.01988) and first released in [this repository](https://github.com/facebookresearch/vissl/tree/main/projects/SEER).
15
+
16
+ Disclaimer: The team releasing RegNetModel did not write a model card for this model so this model card has been written by the Hugging Face team.
17
+
18
+ ## Model description
19
+
20
+ This gigantic model is a scale up [RegNetY](https://arxiv.org/abs/2003.13678) model trained on one bilion random images.
21
+
22
+ ![model image](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/regnet_architecture.png)
23
+
24
+ ## Intended uses & limitations
25
+
26
+ You can use the raw model for image classification. See the [model hub](https://huggingface.co/models?search=regnet) to look for
27
+ fine-tuned versions on a task that interests you.
28
+
29
+ ### How to use
30
+
31
+ Here is how to use this model:
32
+
33
+ ```python
34
+ >>> from transformers import AutoFeatureExtractor, RegNetModel
35
+ >>> import torch
36
+ >>> from datasets import load_dataset
37
+
38
+ >>> dataset = load_dataset("huggingface/cats-image")
39
+ >>> image = dataset["test"]["image"][0]
40
+
41
+ >>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/regnet-y-040")
42
+ >>> model = RegNetModel.from_pretrained("facebook/regnet-y-040")
43
+
44
+ >>> inputs = feature_extractor(image, return_tensors="pt")
45
+
46
+ >>> with torch.no_grad():
47
+ ... outputs = model(**inputs)
48
+
49
+ >>> last_hidden_states = outputs.last_hidden_state
50
+ >>> list(last_hidden_states.shape)
51
+ [1, 1088, 7, 7]
52
+ ```
53
+
54
+
55
+
56
+ For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/regnet).