diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index a575f07..fa2340b 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -360,7 +360,10 @@ def pre_processing_image(image_url): y_offset = (target_height - new_height) // 2 # 将调整大小后的图片粘贴到透明图片上 - result_image.paste(resized_image, (x_offset, y_offset), mask=resized_image.split()[3]) + if resized_image.mode == "RGBA": + result_image.paste(resized_image, (x_offset, y_offset), mask=resized_image.split()[3]) + else: + result_image.paste(resized_image, (x_offset, y_offset)) image = np.array(result_image)