LiheYoung / Depth-Anything

[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation
https://depth-anything.github.io
Apache License 2.0
6.84k stars 525 forks source link

metric_depth模型怎么使用tensorrt进行推理 #187

Open gopin95 opened 3 months ago

gopin95 commented 3 months ago

仓库中的metric_depth模型怎么使用tensorrt进行推理?

lin-voxvision commented 3 months ago

same question, how can we convert the depth_anything_metric_depth_outdoor.pt to an onnx file and then convert it to a trt engine file?

pinnintipraneethkumar commented 3 months ago

Hi, @gopin95 @lin-voxvision , you can follow this github page to use depth-anything v2 with tensorrt https://github.com/spacewalk01/depth-anything-tensorrt

gopin95 commented 3 months ago

Does v2 exist a metric depth estimation model?

pinnintipraneethkumar commented 3 months ago

yes, Depth-Anything v2 have metric depth estimation model , https://github.com/DepthAnything/Depth-Anything-V2/tree/main/metric_depth

lin-voxvision commented 3 months ago

Hi, @gopin95 @lin-voxvision , you can follow this github page to use depth-anything v2 with tensorrt https://github.com/spacewalk01/depth-anything-tensorrt

Seems this tensorrt-convert repo only support relative depth model, but I'll try metric anyway. and results will be updated later, Thanks!!