Model card updates and fixes
Browse files
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](
|
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-
|
46 |
|
47 |
```bash
|
48 |
-
|
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-
|
57 |
-
--include "
|
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.
|