aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat generate 污点图保存

Browse Source
This commit is contained in:
zhouchengrong
2024-05-13 10:44:20 +08:00
parent 36677b1a86
commit a4c318310b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/service/generate_image/service.py
View File

@@ -110,7 +110,8 @@ class GenerateImage:
self.generate_data['message'] = "success" self.generate_data['message'] = "success"
self.generate_data['image_url'] = str(image_url) self.generate_data['image_url'] = str(image_url)
self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) 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['status'] = "SUCCESS"
self.generate_data['message'] = "success" self.generate_data['message'] = "success"
self.generate_data['image_url'] = str(GI_SYS_IMAGE_URL) self.generate_data['image_url'] = str(GI_SYS_IMAGE_URL)