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 的子文件夾中）。