Depth Anything Small Hf

🚀 深度估計模型項目

本項目將LiheYoung/depth-anything-small-hf模型的ONNX權重進行處理，使其與Transformers.js兼容，可用於深度估計任務。

🚀 快速開始

本項目使用 Transformers.js JavaScript庫實現深度估計功能，你可以按照以下步驟快速開始使用。

📦 安裝指南

如果你還未安裝 Transformers.js，可以使用以下命令從 NPM 進行安裝：

npm i @huggingface/transformers

💻 使用示例

基礎用法

以下是使用 Xenova/depth-anything-small-hf 進行深度估計的示例代碼：

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

// 創建深度估計管道
const depth_estimator = await pipeline('depth-estimation', 'Xenova/depth-anything-small-hf');

// 預測給定圖像的深度圖
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/bread_small.png';
const output = await depth_estimator(url);
// {
//   predicted_depth: Tensor {
//     dims: [350, 518],
//     type: 'float32',
//     data: Float32Array(181300) [...],
//     size: 181300
//   },
//   depth: RawImage {
//     data: Uint8Array(271360) [...],
//     width: 640,
//     height: 424,
//     channels: 1
//   }
// }

高級用法

你可以使用以下代碼將輸出的深度圖保存為圖片：

output.depth.save('depth.png');

⚠️ 重要提示

目前為ONNX權重單獨創建一個倉庫是一個臨時解決方案，直到WebML獲得更多關注。如果你想讓你的模型適用於Web，我們建議使用 🤗 Optimum 將模型轉換為ONNX格式，並像本倉庫一樣組織你的倉庫結構（將ONNX權重放在名為 onnx 的子文件夾中）。