coreml-projects/DepthPro-coreml-pruned-10-quantized-linear

This repo contains 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

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-pruned-10-quantized-linear \
  --include "DepthProPruned10QuantizedLinear.mlpackage/*""

To download everything, skip the --include argument.

Conversion Tutorial

The huggingface/coreml-examples repository contains sample conversion code for DepthProPruned10QuantizedLinear.mlpackage and other models.

Swift Integration

The huggingface/coreml-examples repository contains sample Swift code for DepthProPruned10QuantizedLinear.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.