From b52c87bd15cc5d4c53f8837ee4f4790ea45499b4 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Sun, 1 Dec 2024 17:39:22 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20product=20=E4=BF=AE=E5=A4=8D=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../generate_image/service_generate_product_image.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index ebffb05..60b19cb 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -55,10 +55,7 @@ class GenerateProductImage: self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) else: # pil图像转成numpy数组 - if self.product_type == "single": - image = result.as_numpy("generated_inpaint_image") - else: - image = result.as_numpy("generated_cnet_image") + image = result.as_numpy("generated_inpaint_image") image_result = Image.fromarray(np.squeeze(image.astype(np.uint8))).resize(self.image_size) cropped_image = post_processing_image(image_result, self.left, self.top) image_url = upload_SDXL_image(cropped_image, user_id=self.user_id, category=f"{self.category}", file_name=f"{self.tasks_id}.png") @@ -99,7 +96,6 @@ class GenerateProductImage: inputs = [input_text, input_image, input_image_strength] - if self.product_type == "single": ctx = self.grpc_client.async_infer(model_name="stable_diffusion_xl_cnet_inpaint", inputs=inputs, callback=self.callback) else: @@ -187,8 +183,8 @@ if __name__ == '__main__': # prompt="", image_strength=0.7, prompt="The best quality, masterpiece,outwear, 8K realistic, HUD", - image_url="aida-results/result_836dce70-ad59-11ef-86ab-0242ac130002.png", - product_type="single" + image_url="aida-results/result_53381ada-ac64-11ef-ae9d-0242ac150002.png", + product_type="overall" ) server = GenerateProductImage(rd) - print(server.get_result()) \ No newline at end of file + print(server.get_result())