| | ---
|
| | tags:
|
| | - depth-estimation
|
| | library_name: coreml
|
| | license: apple-ascl
|
| | base_model:
|
| | - apple/DepthPro
|
| | ---
|
| |
|
| | This repo contains [DepthProPruned10QuantizedLinear.mlpackage](DepthProPruned10QuantizedLinear.mlpackage)(1100 MB).
|
| |
|
| | This model was first pruned to 10% sparsity, then the weights were linearly quantized.
|
| |
|
| | # DepthPro CoreML Models
|
| |
|
| | DepthPro is a monocular depth estimation model. This means that it is trained to predict depth on a single image.
|
| |
|
| | [DepthPro paper](https://arxiv.org/pdf/2410.02073)
|
| |
|
| | [DepthPro original repo](https://huggingface.co/apple/DepthPro)
|
| |
|
| | # Model Inputs and Outputs
|
| |
|
| | ### Inputs
|
| |
|
| | - `image`: $1536 \times 1536$ 3 color image ($[1 \times 3 \times 1536 \times 1536]$ ImageType).
|
| | - `originalWidth`: A scalar containing the original width of the image before resizing ($[1 \times 1 \times 1 \times 1]$ TensorType).
|
| |
|
| | ### Outputs
|
| |
|
| | - `depthMeters`: $1536 \times 1536$ 1 channel tensor containing depth in meters ($[1 \times 1 \times 1536 \times 1536]$ Tensor).
|
| |
|
| | # Download
|
| |
|
| | Install `huggingface-cli`
|
| |
|
| | ```bash
|
| | brew install huggingface-cli
|
| | ```
|
| |
|
| | To download:
|
| |
|
| | ```bash
|
| | huggingface-cli download \
|
| | --local-dir models --local-dir-use-symlinks False \
|
| | coreml-projects/DepthPro-coreml-pruned-10-quantized-linear \
|
| | --include "DepthProPruned10QuantizedLinear.mlpackage/*""
|
| | ```
|
| |
|
| | To download everything, skip the `--include` argument.
|
| |
|
| | # Conversion Tutorial
|
| |
|
| | The [`huggingface/coreml-examples`](https://github.com/huggingface/coreml-examples/blob/main/tutorials/DepthPro/depth_pro_coreml_guide.ipynb) repository contains sample conversion code for `DepthProPruned10QuantizedLinear.mlpackage` and other models.
|
| |
|
| | # Swift Integration
|
| |
|
| | The [`huggingface/coreml-examples`](https://github.com/huggingface/coreml-examples/blob/main/DepthProSample/README.md) repository contains sample Swift code for `DepthProPruned10QuantizedLinear.mlpackage` and other models. See [the instructions there](https://github.com/huggingface/coreml-examples/tree/main/DepthProSample) to build the demo app, which shows how to use the model in your own Swift apps. |
