LBM:潜在桥接匹配用于图像重照明

潜在桥匹配(LBM)是一种新颖、多功能且可扩展的方法,在《LBM:用于快速图像到图像转换的潜在桥匹配》中提出,该方法依赖于潜在空间中的桥匹配来实现快速的图像到图像转换。该模型经过训练,可以根据提供的背景重新照亮前景对象。请查看我们的实时演示和官方Github仓库

在这里插入图片描述

如何使用?

要使用此模型,您首先需要通过运行以下命令安装相关的 lbm 库

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

然后,你可以在输入图像上使用模型进行推理。

import torch
from diffusers.utils import load_image
from lbm.inference import evaluate, get_model

# Load model
model = get_model(
  "jasperai/LBM_relighting",
  torch_dtype=torch.bfloat16,
  device="cuda",
)

# Load a source image
source_image = load_image(
  "https://huggingface.co/jasperai/LBM_relighting/resolve/main/assets/source_image.jpg"
)

# Perform inference
output_image = evaluate(model, source_image, num_sampling_steps=1)
output_image

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值