pcuenq HF staff commited on
Commit
0c3166e
·
1 Parent(s): bd2b1cf

Model card updates and fixes

Browse files
Files changed (1) hide show
  1. README.md +13 -8
README.md CHANGED
@@ -1,15 +1,16 @@
1
  ---
 
 
 
2
  license: apache-2.0
3
  ---
4
 
5
  # DETR-Resnet50 (semantic segmentation) Core ML Models
6
 
7
- See [the Files tab](https://huggingface.co/coreml-projects/detr-resnet50-semantic-segmentation/tree/main) for converted models.
8
 
9
  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).
10
 
11
- 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.
12
-
13
  ## Model description
14
 
15
  The DETR model is an encoder-decoder transformer with a convolutional backbone. Two heads are added on top of the decoder outputs in order to perform object detection: a linear layer for the class labels and a MLP (multi-layer perceptron) for the bounding boxes. The model uses so-called object queries to detect objects in an image. Each object query looks for a particular object in the image. For COCO, the number of object queries is set to 100.
@@ -42,10 +43,10 @@ The following results refer to DETRResnet50SemanticSegmentationF16. The compute
42
 
43
  ## Download
44
 
45
- Install `huggingface-hub`
46
 
47
  ```bash
48
- pip install huggingface-hub
49
  ```
50
 
51
  To download one of the `.mlpackage` folders to the `models` directory:
@@ -53,8 +54,12 @@ To download one of the `.mlpackage` folders to the `models` directory:
53
  ```bash
54
  huggingface-cli download \
55
  --local-dir models --local-dir-use-symlinks False \
56
- coreml-projects/detr-resnet50-semantic-segmentation \
57
- --include "detr-resnet50-semantic-400-float16.mlpackage/*"
58
  ```
59
 
60
- To download everything, skip the `--include` argument.
 
 
 
 
 
1
  ---
2
+ tags:
3
+ - image-segmentation
4
+ library_name: coreml
5
  license: apache-2.0
6
  ---
7
 
8
  # DETR-Resnet50 (semantic segmentation) Core ML Models
9
 
10
+ See [the Files tab](tree/main) for converted models.
11
 
12
  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).
13
 
 
 
14
  ## Model description
15
 
16
  The DETR model is an encoder-decoder transformer with a convolutional backbone. Two heads are added on top of the decoder outputs in order to perform object detection: a linear layer for the class labels and a MLP (multi-layer perceptron) for the bounding boxes. The model uses so-called object queries to detect objects in an image. Each object query looks for a particular object in the image. For COCO, the number of object queries is set to 100.
 
43
 
44
  ## Download
45
 
46
+ Install `huggingface-cli`
47
 
48
  ```bash
49
+ brew install huggingface-cli
50
  ```
51
 
52
  To download one of the `.mlpackage` folders to the `models` directory:
 
54
  ```bash
55
  huggingface-cli download \
56
  --local-dir models --local-dir-use-symlinks False \
57
+ apple/coreml-detr-semantic-segmentation \
58
+ --include "DETRResnet50SemanticSegmentationF16.mlpackage/*"
59
  ```
60
 
61
+ To download everything, skip the `--include` argument. This will retrieve `float32` and `float16` variants, as well as quantized versions of the `float16` variant.
62
+
63
+ ## Integrate in Swift apps
64
+
65
+ The [`huggingface/coreml-examples`](https://github.com/huggingface/coreml-examples/blob/main/depth-anything-example/README.md) repository contains sample Swift code for `coreml-detr-semantic-segmentation` and other models. See [the instructions there](https://github.com/huggingface/coreml-examples/blob/main/SemanticSegmentationSample/README.md) to build the demo app, which shows how to use the model in your own Swift apps.