Depth Anything V2 Small

🚀 Depth-Anything-V2-Small と Transformers.js を用いた深度推定

このプロジェクトは、Depth-Anything-V2-Small モデルを ONNX 形式に変換し、Transformers.js と互換性を持たせることで、Web 上での深度推定を可能にします。

🚀 クイックスタート

ライブラリのインストール

まだの場合は、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');

📚 詳細情報

ONNX 重みについて

ONNX 重みを別のリポジトリにするのは、WebML が普及するまでの一時的な解決策です。モデルを Web 対応にする場合は、🤗 Optimum を使って ONNX に変換し、このリポジトリのように構成することをおすすめします（ONNX 重みは onnx というサブフォルダに配置）。

📄 ライセンス

このプロジェクトは Apache-2.0 ライセンスの下で公開されています。