Depth Anything V2 Small

🚀 Depth-Anything-V2-Small

Depth Anything V2は、595Kの合成ラベル付き画像と62M以上の実際のラベルなし画像から学習された、最も高性能な単眼深度推定（MDE）モデルを提供します。以下の特徴を備えています。

• Depth Anything V1よりも細かい詳細を捉えることができます。
• Depth Anything V1やSDベースのモデル（例：Marigold、Geowizard）よりも堅牢です。
• SDベースのモデルよりも10倍高速で軽量です。
• 事前学習モデルを用いた微調整で印象的な性能を発揮します。

🚀 クイックスタート

このセクションでは、Depth-Anything-V2-Smallのクイックスタートについて説明します。

✨ 主な機能

• Depth Anything V1よりも細かい詳細を捉えることができます。
• Depth Anything V1やSDベースのモデル（例：Marigold、Geowizard）よりも堅牢です。
• SDベースのモデルよりも10倍高速で軽量です。
• 事前学習モデルを用いた微調整で印象的な性能を発揮します。

📦 インストール

以下のコマンドを実行して、Depth-Anything-V2をインストールします。

git clone https://huggingface.co/spaces/depth-anything/Depth-Anything-V2
cd Depth-Anything-V2
pip install -r requirements.txt

💻 使用例

基本的な使用法

まず、モデルをダウンロードし、checkpointsディレクトリに配置します。

import cv2
import torch

from depth_anything_v2.dpt import DepthAnythingV2

model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location='cpu'))
model.eval()

raw_img = cv2.imread('your/image/path')
depth = model.infer_image(raw_img) # HxW raw depth map

📚 ドキュメント

このセクションでは、Depth-Anything-V2-Smallに関する詳細なドキュメントを提供します。

📄 ライセンス

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

📚 引用

このプロジェクトが役に立った場合は、以下の文献を引用してください。

@article{depth_anything_v2,
  title={Depth Anything V2},
  author={Yang, Lihe and Kang, Bingyi and Huang, Zilong and Zhao, Zhen and Xu, Xiaogang and Feng, Jiashi and Zhao, Hengshuang},
  journal={arXiv:2406.09414},
  year={2024}
}

@inproceedings{depth_anything_v1,
  title={Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data}, 
  author={Yang, Lihe and Kang, Bingyi and Huang, Zilong and Xu, Xiaogang and Feng, Jiashi and Zhao, Hengshuang},
  booktitle={CVPR},
  year={2024}
}