Depth Anything V2 Small

🚀 深度估计模型

本项目基于 depth-anything/Depth-Anything-V2-Small 基础模型，借助 transformers.js 库实现深度估计功能，可有效解决图像深度估计的问题，为相关应用提供便捷的深度信息获取方案。

🚀 快速开始

本项目使用来自 Hugging Face 带有 ONNX 权重的模型，以兼容 Transformers.js 库。

📦 安装指南

若你还未安装 Transformers.js JavaScript 库，可以通过以下命令从 NPM 进行安装：

npm i @huggingface/transformers

💻 使用示例

基础用法

以下是使用 onnx-community/depth-anything-v2-small 进行深度估计的示例代码：

import { pipeline } from '@huggingface/transformers';

// 创建深度估计管道
const depth_estimator = await pipeline('depth-estimation', 'onnx-community/depth-anything-v2-small');

// 预测图像的深度
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/cats.jpg';
const { depth } = await depth_estimator(url);

// 可视化输出
depth.save('depth.png');

📄 许可证

本项目采用 apache-2.0 许可证。

⚠️ 重要提示

目前为 ONNX 权重单独创建一个仓库是临时解决方案，直到 WebML 获得更多关注。如果你想让你的模型适用于 Web，建议使用 🤗 Optimum 将模型转换为 ONNX 格式，并像本仓库一样组织你的仓库结构（将 ONNX 权重放在名为 onnx 的子文件夹中）。