Depth Pro How to use aarondevstack/DepthPro-512x512-coreml with Depth Pro:
# Download checkpoint
pip install huggingface-hub
huggingface-cli download --local-dir checkpoints aarondevstack/DepthPro-512x512-coreml
import depth_pro
# Load model and preprocessing transform
model, transform = depth_pro.create_model_and_transforms()
model.eval()
# Load and preprocess an image.
image, _, f_px = depth_pro.load_rgb("example.png")
image = transform(image)
# Run inference.
prediction = model.infer(image, f_px=f_px)
# Results: 1. Depth in meters
depth = prediction["depth"]
# Results: 2. Focal length in pixels
focallength_px = prediction["focallength_px"]