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

feat flux 取消污点检测 增加类别判断

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-12-01 19:52:33 +08:00
parent 87d3848088
commit 0f17d8b654
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/generate_image/service_generate_image.py
View File

@@ -91,7 +91,7 @@ class GenerateImage:
image = result.as_numpy("generated_image") image = result.as_numpy("generated_image")
image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR) image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR)
is_smudge = True is_smudge = True
if self.category == "sketch" and self.version == "fast": if self.category == "sketch":
if self.version == "fast": if self.version == "fast":
# 色阶调整 # 色阶调整
cutoff = 1 cutoff = 1