Depth Anything Base Hf

🚀 Depth Anything Base HF（Transformers.js版）

このプロジェクトは、LiheYoung/depth-anything-base-hfモデルをTransformers.jsと互換性のあるONNXウェイトで提供しています。深度推定タスクに最適です。

🚀 クイックスタート

📦 インストール

まだの場合は、Transformers.js JavaScriptライブラリをNPMから以下のコマンドでインストールできます。

npm i @huggingface/transformers

💻 使用例

基本的な使用法

// インポート
import { pipeline } from '@huggingface/transformers';

// 深度推定パイプラインの作成
const depth_estimator = await pipeline('depth-estimation', 'Xenova/depth-anything-base-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というサブフォルダに配置）。

情報一覧