LBM Depth

🚀 ✨ 用於深度估計的潛在橋接匹配 ✨

潛在橋接匹配（Latent Bridge Matching，LBM）是一種全新的、通用且可擴展的方法，該方法在論文 LBM: Latent Bridge Matching for Fast Image-to-Image Translation 中被提出。它藉助潛在空間中的橋接匹配來實現快速的圖像到圖像轉換。此模型經過訓練，可根據給定的輸入圖像估計深度圖。你還可以查看我們用於圖像重光照的 即時演示 以及官方 Github 倉庫。

🚀 快速開始

安裝相關庫

要使用此模型，你首先需要通過運行以下命令來安裝相關的 lbm 庫：

pip install git+https://github.com/gojasper/LBM.git

對輸入圖像進行推理

安裝完成後，你可以使用該模型對輸入圖像進行推理：

import torch
from PIL import Image
from lbm.inference import evaluate, get_model

# 加載模型
model = get_model(
  "jasperai/LBM_relighting",
  torch_dtype=torch.bfloat16,
  device="cuda",
)

# 加載源圖像
source_image = Image.open("your_image")

# 進行推理
output_image = evaluate(model, source_image, num_sampling_steps=1)

📊 指標

該模型取得了以下指標：

📄 許可證

此代碼是在 知識共享署名 - 非商業性使用 4.0 國際許可協議（Creative Commons BY - NC 4.0） 下發布的。

📚 引用

如果你覺得這項工作有用，或者在你的研究中使用了它，請考慮引用我們：

@article{chadebec2025lbm,
      title={LBM: Latent Bridge Matching for Fast Image-to-Image Translation}, 
      author={Clément Chadebec and Onur Tasar and Sanjeev Sreetharan and Benjamin Aubin},
      year={2025},
      journal = {arXiv preprint arXiv:2503.07535},
}