| tags: | |
| - depth-estimation | |
| library_name: coreml | |
| license: apple-ascl | |
| base_model: | |
| - apple/DepthPro | |
| This repo contains [DepthPro.mlpackage](DepthPro.mlpackage)(1900 MB). | |
| # 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 \ | |
| --include "DepthPro.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 `DepthPro.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 `DepthPro.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. |