26.10.2023 в 09:22
Алексей Устинов
Замените любой объект на фотографии с помощью Stable Diffusion Python SDK
Вы можете установить Novita Python SDK с помощью всего одной команды.
pip install novita-clientполучите NOVITA_API_KEY из https://novita.ai/get-started/
Наконец, вы можете попробовать сменить кошку на собаку, следуя этому коду.
import os
from novita_client import NovitaClient, Txt2ImgRequest, UpscaleRequest
from novita_client.utils import base64_to_image, image_to_base64
from PIL import Image
from io import BytesIO
novita = NovitaClient(os.getenv('NOVITA_API_KEY'))
replace_object_res = novita.replace_object(image="https://cdn-images-1.medium.com/max/1600/1*E3jkiGoC8EfVlqqhE_6SWw.png", object_prompt="a cat", prompt="a dog", response_image_type="png")
replace_object_generated = base64_to_image(replace_object_res.image_file)
replace_object_generated.save("replace-object.png")
novita-client — это Python SDK Novita AI, который предоставляет следующие функции.
- cleanup — удалит все сгенерированные изображения
- remove-background — удалит фон с изображения
- remove-text — удалит текст с изображения
- reimagine – переосмыслит образ
- doodle — изображение каракулей
- merge-face — объединит лицо
- mix-pose — смешанная поза
- outpainting — перекраска
- replace-object — удалит объект
- replace-background — заменит фон
- replace-sky — заменит небо
- create-tile — создаст плитку
