nielsr HF staff commited on
Commit
f7bfb7d
1 Parent(s): 7bcdcde

Update model card

Browse files
Files changed (1) hide show
  1. README.md +4 -10
README.md CHANGED
@@ -3,21 +3,15 @@ license: apache-2.0
3
  tags:
4
  ---
5
 
6
- # Vision Transformer (base-sized model)
7
 
8
- Vision Transformer (ViT) model pre-trained on ImageNet-21k (14 million images, 21,843 classes) at resolution 224x224. It was introduced in the paper [An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale](https://arxiv.org/abs/2010.11929) by Dosovitskiy et al. and first released in [this repository](https://github.com/google-research/vision_transformer). However, the weights were converted from the [timm repository](https://github.com/rwightman/pytorch-image-models) by Ross Wightman, who already converted the weights from JAX to PyTorch. Credits go to him.
9
 
10
- Disclaimer: The team releasing ViT did not write a model card for this model so this model card has been written by the Hugging Face team.
11
 
12
  ## Model description
13
 
14
- The Vision Transformer (ViT) is a transformer encoder model (BERT-like) pretrained on a large collection of images in a supervised fashion, namely ImageNet-21k, at a resolution of 224x224 pixels.
15
-
16
- Images are presented to the model as a sequence of fixed-size patches (resolution 16x16), which are linearly embedded. One also adds a [CLS] token to the beginning of a sequence to use it for classification tasks. One also adds absolute position embeddings before feeding the sequence to the layers of the Transformer encoder.
17
-
18
- Note that this model does not provide any fine-tuned heads, as these were zero'd by Google researchers. However, the model does include the pre-trained pooler, which can be used for downstream tasks (such as image classification).
19
-
20
- By pre-training the model, it learns an inner representation of images that can then be used to extract features useful for downstream tasks: if you have a dataset of labeled images for instance, you can train a standard classifier by placing a linear layer on top of the pre-trained encoder. One typically places a linear layer on top of the [CLS] token, as the last hidden state of this token can be seen as a representation of an entire image.
21
 
22
  ## Intended uses & limitations
23
 
 
3
  tags:
4
  ---
5
 
6
+ # DETR (End-to-End Object Detection) model with ResNet-50 backbone
7
 
8
+ DEtection Transformer (DETR) model trained end-to-end on COCO 2017 object detection (118k annotated images). It was introduced in the paper [End-to-End Object Detection with Transformers](https://arxiv.org/abs/2005.12872) by Carion et al. and first released in [this repository](https://github.com/facebookresearch/detr).
9
 
10
+ Disclaimer: The team releasing DETR did not write a model card for this model so this model card has been written by the Hugging Face team.
11
 
12
  ## Model description
13
 
14
+ The DETR model is an encoder-decoder transformer with a convolutional backbone.
 
 
 
 
 
 
15
 
16
  ## Intended uses & limitations
17