Depth-Anything-V2

此文档讲解如何在安装了瑞莎智核 AX-M1 的 host 设备上运行 Depth-Anything-V2 示例应用。

预编译模型量化方式：w8a16

下载示例应用仓库

使用 huggingfcae-cli 下载示例应用仓库。

Host

pip3 install -U "huggingface_hub[cli]"
huggingface-cli download AXERA-TECH/Depth-Anything-V2 --local-dir ./Depth-Anything-V2
cd Depth-Anything-V2

示例使用

安装 python 依赖

Host

pip3 install argparse numpy opencv-python
pip3 install https://github.com/AXERA-TECH/pyaxengine/releases/download/0.1.3.rc1/axengine-0.1.3-py3-none-any.whl

模型推理

Host

python3 python/infer.py --model depth_anything_v2_vits_ax650.axmodel --img examples/demo01.jpg

(.venv) rock@rock-5b-plus:~/ssd/axera/Depth-Anything-V2$ python3 python/infer.py --model depth_anything_v2_vits_ax650.axmodel --img examples/demo01.jpg
[INFO] Available providers:  ['AXCLRTExecutionProvider']
[INFO] Using provider: AXCLRTExecutionProvider
[INFO] SOC Name: AX650N
[INFO] VNPU type: VNPUType.DISABLED

depth-anything-v2 demo output