feat(新功能):

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试): 旧版product 测试
This commit is contained in:
zhouchengrong
2025-02-03 19:06:49 +08:00
parent 07e09dc99d
commit e1c00bbd66

View File

@@ -334,17 +334,18 @@ def pre_processing_image(image_url):
resized_image = image.resize((new_width, new_height))
# 创建一个512x768的透明图片
result_image = Image.new("RGBA", (512, 768), (0, 0, 0, 0))
result_image = Image.new("RGBA", (512, 768), (255, 255, 255, 255))
# 计算需要粘贴的位置,使图片居中
x_offset = (512 - new_width) // 2
y_offset = 0
# 将调整大小后的图片粘贴到透明图片上
result_image.paste(resized_image, (x_offset, y_offset))
result_image.paste(resized_image, (x_offset, y_offset), mask=resized_image.split()[3])
image = np.array(result_image)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
# image = cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA)
return image
@@ -366,8 +367,8 @@ if __name__ == '__main__':
tasks_id="123-89",
# prompt="",
image_strength=0.7,
prompt="The best quality, masterpiece,outwear, 8K realistic, HUD",
image_url="aida-results/result_53381ada-ac64-11ef-ae9d-0242ac150002.png",
prompt="The best quality, masterpiece, real image.,high quality clothing details,8K realistic,HDR",
image_url="aida-users/11633/toProductImageElement/46166c36-c584-4e0f-b9fe-50615ec03ef3.png",
product_type="overall"
)
server = GenerateProductImage(rd)