davanstrien HF staff commited on
Commit
fce50e7
1 Parent(s): 0a1bdfc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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