feat generate 污点图保存

2024-05-13 10:44:20 +08:00
parent 36677b1a86
commit a4c318310b
1 changed files with 2 additions and 1 deletions
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -110,7 +110,8 @@ class GenerateImage:
                self.generate_data['message'] = "success"
                self.generate_data['image_url'] = str(image_url)
                self.redis_client.set(self.tasks_id, json.dumps(self.generate_data))
-            else:  # 有污点
+            else:  # 有污点 保存图片到本地 测试用
+                cv2.imwrite(f"generate_result/{self.tasks_id}.png", image_result)
                self.generate_data['status'] = "SUCCESS"
                self.generate_data['message'] = "success"
                self.generate_data['image_url'] = str(GI_SYS_IMAGE_URL)