CoreML DepthPro
Collection
4 items
•
Updated
This repo contains DepthPro.mlpackage(1900 MB).
DepthPro is a monocular depth estimation model. This means that it is trained to predict depth on a single image.
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).depthMeters
: $1536 \times 1536$ 1 channel tensor containing depth in meters ($[1 \times 1 \times 1536 \times 1536]$ Tensor).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.
The huggingface/coreml-examples
repository contains sample conversion code for DepthPro.mlpackage
and other models.
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.
Base model
apple/DepthPro