metadata
tags:
- depth-estimation
library_name: coreml
license: apple-ascl
base_model:
- apple/DepthPro
This repo contains 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.
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
brew install huggingface-cli
To download:
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
repository contains sample conversion code for DepthPro.mlpackage
and other models.
Swift Integration
The huggingface/coreml-examples
repository contains sample Swift code for DepthPro.mlpackage
and other models. See the instructions there to build the demo app, which shows how to use the model in your own Swift apps.