Обнимающая модель диффузии лица - добавление объекта к существующей фотографии ⇐ Python

[Anonymous]

Я играю с библиотекой объятий. Я хотел добавить объект к существующей фотографии. Я предоставил диффузионный трубопровод и попросил его добавить объект. Результатом стала просто оригинальная фотография с некоторыми шумами, ничто не выглядит так, как добавлена ​​палочка золотой цепи. Чтобы достичь того, что я хочу?
from diffusers import StableDiffusionImg2ImgPipeline
import torch
from PIL import Image
import matplotlib.pyplot as plt

# Load img2img pipeline
device = "mps" if torch.backends.mps.is_available() else "cpu"
pipe = StableDiffusionImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-2-1").to(device)

# Load your photo
init_image = Image.open("/Users/myUser/Downloads/Meelo1.png").convert("RGB")
new_width = 512
idth_percent = (new_width / float(init_image.size[0])) # Get the width percentage
new_height = int((float(init_image.size[1]) * float(width_percent)))
resized_image = init_image.resize((new_width, new_height), Image.Resampling.LANCZOS)
# Generate a new image based on your input photo
prompt = "A dog face wearing a thick gold chain"

image = pipe(prompt=prompt, image=resized_image, strength=0.3, num_inference_steps=200).images[0]
plt.imshow(image)
plt.axis("off")
plt.show()


Подробнее здесь: https://stackoverflow.com/questions/794 ... ting-photo