Commit
•
fce50e7
1
Parent(s):
0a1bdfc
Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
19 |
|
20 |
# detr-resnet-50_fine_tuned_nls_chapbooks
|
21 |
|
22 |
-
This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the nls_chapbook_illustrations dataset.
|
23 |
|
24 |
## Model description
|
25 |
|
@@ -29,7 +29,7 @@ More information needed
|
|
29 |
|
30 |
### Using in a transformer pipeline
|
31 |
|
32 |
-
The easiest way to use this model is via a [Transformers pipeline](https://huggingface.co/docs/transformers/main/en/pipeline_tutorial#vision-pipeline). To do this you should first load the model and feature extractor:
|
33 |
|
34 |
```python
|
35 |
from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
|
@@ -39,7 +39,7 @@ extractor = AutoFeatureExtractor.from_pretrained("davanstrien/detr-resnet-50_fin
|
|
39 |
model = AutoModelForObjectDetection.from_pretrained("davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
|
40 |
```
|
41 |
|
42 |
-
Then you can create a pipeline for object detection using the model
|
43 |
|
44 |
```python
|
45 |
from transformers import pipeline
|
|
|
19 |
|
20 |
# detr-resnet-50_fine_tuned_nls_chapbooks
|
21 |
|
22 |
+
This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the `biglam/nls_chapbook_illustrations` dataset. This dataset contains images of chapbooks with bounding boxes for the illustrations contained on some of the pages.
|
23 |
|
24 |
## Model description
|
25 |
|
|
|
29 |
|
30 |
### Using in a transformer pipeline
|
31 |
|
32 |
+
The easiest way to use this model is via a [Transformers pipeline](https://huggingface.co/docs/transformers/main/en/pipeline_tutorial#vision-pipeline). To do this, you should first load the model and feature extractor:
|
33 |
|
34 |
```python
|
35 |
from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
|
|
|
39 |
model = AutoModelForObjectDetection.from_pretrained("davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
|
40 |
```
|
41 |
|
42 |
+
Then you can create a pipeline for object detection using the model.
|
43 |
|
44 |
```python
|
45 |
from transformers import pipeline
|