Depth Estimation
Core ML
DepthPro-coreml / README.md
KeighBee's picture
KeighBee HF staff
added model to repo and updated readme
e3b374e
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.

DepthPro paper

DepthPro original repo

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.