Nova Canvas - 物体替换

from random import randint
import json
import base64
import boto3

client = boto3.client(service_name='bedrock-runtime', region_name="us-east-1")

source_image_path = "three_pots.jpg"

# Read image from disk.
with open(source_image_path, "rb") as image_file:
    source_image_base64 = base64.b64encode(image_file.read()).decode("utf8")


inference_params = json.dumps({
    "taskType": "INPAINTING",
    "inPaintingParams": {
        "image": source_image_base64,
        "maskPrompt": "flowers in pots",  # Description of the elements to replace
        "text": "yellow flowers"
    },
    "imageGenerationConfig": {
        "numberOfImages": 1,  # Number of variations to generate. 1 to 5.
        "quality": "standard",  # Allowed values are "standard" and "premium"
        "cfgScale": 8.0,  # How closely the prompt will be followed
        "seed": randint(0, 858993459),  # Use a random seed
    },
})

response = client.invoke_model(
    body=inference_params,
    modelId="amazon.nova-canvas-v1:0",
    accept="application/json",
    contentType="application/json",
)

response_body = json.loads(response.get("body").read())  # 读取并解析响应体
base64_image = response_body.get("images")[0]  # 获取返回的 Base64 编码图片数据

# 将 Base64 编码的图片解码为二进制数据
base_64_image = base64.b64decode(base64_image)

# 将解码后的图片保存为文件
file_path = "1.png"
with open(file_path, "wb") as f:
    f.write(base_64_image)

print(f"图片已成功保存为: {file_path}")