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

feat generate sketch 污点检测改为顶部两角

Browse Source
This commit is contained in:
zhouchengrong
2024-04-25 15:54:17 +08:00
parent 78d0eb7373
commit a0c47c3367
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -78,7 +78,10 @@ class GenerateImage:
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:
image = result.as_numpy("generated_image") image = result.as_numpy("generated_image")
image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_BGR2RGB) if self.mode == "img2img":
image_result = np.squeeze(image.astype(np.uint8))
else:
image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_BGR2RGB)
is_smudge = True is_smudge = True
if self.category == "sketch": if self.category == "sketch":
# 去背景 # 去背景