标签:

• 深度估计 库名称: coreml 许可证: apple-ascl 基础模型:
  • apple/DepthPro

DepthPro CoreML 模型

DepthPro 是一种单目深度估计模型。这意味着它经过训练，能够通过单张图像预测深度。

DepthPro 论文

DepthPro 原始仓库

模型变体

模型输入与输出

DepthPro 归一化逆深度模型

输入

• image: 1536x1536 3通道彩色图像。

输出

• normalizedInverseDepth: 1536x1536 单通道图像。

DepthPro 模型

输入

• image: 1536x1536 3通道彩色图像。
• originalWidth: 1x1x1x1 张量，包含图像调整大小前的原始宽度。

输出

• depthMeters: 1x1x1536x1536 张量，包含以米为单位的深度值。

下载

安装 huggingface-cli

brew install huggingface-cli

将 .mlpackage 文件夹下载到 models 目录：

huggingface-cli download \
  --local-dir models --local-dir-use-symlinks False \
  KeighBee/coreml-DepthPro \
  --include "DepthProNormalizedInverseDepthPruned10QuantizedLinear.mlpackage/*" "DepthProPruned10QuantizedLinear.mlpackage/*"

若要下载全部内容，请省略 --include 参数。

在 Swift 应用中集成

huggingface/coreml-examples 仓库包含针对 DepthProNormalizedInverseDepthPruned10QuantizedLinear.mlpackage 及其他模型的 Swift 示例代码。参考该处说明构建演示应用，了解如何在您自己的 Swift 应用中使用该模型。