From e1c00bbd669a00028003e0f0388786714399ae64 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 3 Feb 2025 19:06:49 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20docs?= =?UTF-8?q?=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refac?= =?UTF-8?q?tor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=E6=B5=8B=E8=AF=95):=20=E6=97=A7=E7=89=88product=20?= =?UTF-8?q?=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../generate_image/service_generate_product_image.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index 681e2b5..0507953 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -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)