Update README.md
Browse files
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 |
|