ComfyUI Depth Anything Tensorrt Custom Node (up to 5x faster)
This repo provides a ComfyUI Custom Node implementation of the Depth-Anything-Tensorrt in Python for ultra fast depth map generation (up to 5x faster)
Note: The following results were benchmarked on FP16 engines inside ComfyUI
Device | Model | Model Input (WxH) | Image Resolution (WxH) | FPS |
---|---|---|---|---|
RTX4090 | Depth-Anything-S | 518x518 | 1280x720 | 35 |
RTX4090 | Depth-Anything-B | 518x518 | 1280x720 | 33 |
RTX4090 | Depth-Anything-L | 518x518 | 1280x720 | 24 |
Navigate to the ComfyUI /custom_nodes
directory
git clone https://github.com/yuvraj108c/ComfyUI-Depth-Anything-Tensorrt.git
cd ./ComfyUI-Depth-Anything-Tensorrt
pip install -r requirements.txt
python export_trt.py
/models/tensorrt/depth-anything
directoryRight Click -> tensorrt -> Depth Anything Tensorrt
08/05/2024
26/04/2024
7/04/2024
30/03/2024
27/03/2024
@inproceedings{depthanything,
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}
}