Views
No views yet
| Base model | depth-anything/Depth-Anything-V2-Large-hf |
| Dataset | Booster stereo (prepared split, metric depth in metres) |
| Loss | Affine-invariant L1 + gradient loss (scale+shift in disparity space) |
| LoRA rank / alpha | 16 / 32 |
| LoRA targets | query, key, value |
| Best val AbsRel | 0.0333 (scale+shift aligned) |
| Epochs trained | 2 |
1from transformers import AutoModelForDepthEstimation, AutoImageProcessor
2from peft import PeftModel
3from PIL import Image
4import torch
5
6processor = AutoImageProcessor.from_pretrained(
7 "depth-anything/Depth-Anything-V2-Large-hf",
8 size={"height": 518, "width": 518},
9)
10base = AutoModelForDepthEstimation.from_pretrained(
11 "depth-anything/Depth-Anything-V2-Large-hf"
12)
13model = PeftModel.from_pretrained(base, "igzi/depth-anything-v2-large-lora-booster")
14model.eval()
15
16image = Image.open("image.jpg").convert("RGB")
17inputs = processor(images=image, return_tensors="pt")
18with torch.no_grad():
19 outputs = model(**inputs)
20depth = outputs.predicted_depth # relative disparity; use scale+shift to get metric depth