nickmuchi commited on
Commit
2c3957e
1 Parent(s): 08860ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md CHANGED
@@ -21,14 +21,20 @@ model-index:
21
  results: []
22
  ---
23
  # YOLOS (small-sized) model
 
24
  The original YOLOS model was fine-tuned on COCO 2017 object detection (118k annotated images). It was introduced in the paper [You Only Look at One Sequence: Rethinking Transformer in Vision through Object Detection](https://arxiv.org/abs/2106.00666) by Fang et al. and first released in [this repository](https://github.com/hustvl/YOLOS).
25
  This model was further fine-tuned on the [license plate dataset]("https://www.kaggle.com/datasets/andrewmvd/car-plate-detection") from Kaggle. The dataset consists of 735 images of annotations categorised as "vehicle" and "license-plate". The model was trained for 200 epochs on a single GPU using Google Colab
 
26
  ## Model description
 
27
  YOLOS is a Vision Transformer (ViT) trained using the DETR loss. Despite its simplicity, a base-sized YOLOS model is able to achieve 42 AP on COCO validation 2017 (similar to DETR and more complex frameworks such as Faster R-CNN).
28
  ## Intended uses & limitations
29
  You can use the raw model for object detection. See the [model hub](https://huggingface.co/models?search=hustvl/yolos) to look for all available YOLOS models.
 
30
  ### How to use
 
31
  Here is how to use this model:
 
32
  ```python
33
  from transformers import YolosFeatureExtractor, YolosForObjectDetection
34
  from PIL import Image
@@ -44,14 +50,22 @@ logits = outputs.logits
44
  bboxes = outputs.pred_boxes
45
  ```
46
  Currently, both the feature extractor and model support PyTorch.
 
47
  ## Training data
 
48
  The YOLOS model was pre-trained on [ImageNet-1k](https://huggingface.co/datasets/imagenet2012) and fine-tuned on [COCO 2017 object detection](https://cocodataset.org/#download), a dataset consisting of 118k/5k annotated images for training/validation respectively.
49
  ### Training
 
50
  This model was fine-tuned for 200 epochs on the [license plate dataset]("https://www.kaggle.com/datasets/andrewmvd/car-plate-detection").
 
51
  ## Evaluation results
 
52
  This model achieves an AP (average precision) of **47.9**.
 
53
  Accumulating evaluation results...
 
54
  IoU metric: bbox
 
55
  Metrics | Metric Parameter | Location | Dets | Value |
56
  ---------------- | --------------------- | ------------| ------------- | ----- |
57
  Average Precision | (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] | 0.479 |
 
21
  results: []
22
  ---
23
  # YOLOS (small-sized) model
24
+
25
  The original YOLOS model was fine-tuned on COCO 2017 object detection (118k annotated images). It was introduced in the paper [You Only Look at One Sequence: Rethinking Transformer in Vision through Object Detection](https://arxiv.org/abs/2106.00666) by Fang et al. and first released in [this repository](https://github.com/hustvl/YOLOS).
26
  This model was further fine-tuned on the [license plate dataset]("https://www.kaggle.com/datasets/andrewmvd/car-plate-detection") from Kaggle. The dataset consists of 735 images of annotations categorised as "vehicle" and "license-plate". The model was trained for 200 epochs on a single GPU using Google Colab
27
+
28
  ## Model description
29
+
30
  YOLOS is a Vision Transformer (ViT) trained using the DETR loss. Despite its simplicity, a base-sized YOLOS model is able to achieve 42 AP on COCO validation 2017 (similar to DETR and more complex frameworks such as Faster R-CNN).
31
  ## Intended uses & limitations
32
  You can use the raw model for object detection. See the [model hub](https://huggingface.co/models?search=hustvl/yolos) to look for all available YOLOS models.
33
+
34
  ### How to use
35
+
36
  Here is how to use this model:
37
+
38
  ```python
39
  from transformers import YolosFeatureExtractor, YolosForObjectDetection
40
  from PIL import Image
 
50
  bboxes = outputs.pred_boxes
51
  ```
52
  Currently, both the feature extractor and model support PyTorch.
53
+
54
  ## Training data
55
+
56
  The YOLOS model was pre-trained on [ImageNet-1k](https://huggingface.co/datasets/imagenet2012) and fine-tuned on [COCO 2017 object detection](https://cocodataset.org/#download), a dataset consisting of 118k/5k annotated images for training/validation respectively.
57
  ### Training
58
+
59
  This model was fine-tuned for 200 epochs on the [license plate dataset]("https://www.kaggle.com/datasets/andrewmvd/car-plate-detection").
60
+
61
  ## Evaluation results
62
+
63
  This model achieves an AP (average precision) of **47.9**.
64
+
65
  Accumulating evaluation results...
66
+
67
  IoU metric: bbox
68
+
69
  Metrics | Metric Parameter | Location | Dets | Value |
70
  ---------------- | --------------------- | ------------| ------------- | ----- |
71
  Average Precision | (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] | 0.479 |