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

feat（新功能）:

Browse Source 
fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试): 旧版product 测试
This commit is contained in:
zhouchengrong
2025-02-03 19:13:48 +08:00
parent e1c00bbd66
commit 74008c4586
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_generate_product_image.py
View File

@@ -263,7 +263,8 @@ class GenerateProductImage:
def get_result(self): def get_result(self):
try: try:
prompts = [self.prompt] * self.batch_size prompts = [self.prompt] * self.batch_size
self.image = cv2.cvtColor(self.image, cv2.COLOR_BGR2RGB)
self.image = cv2.cvtColor(self.image, cv2.COLOR_RGBA2RGB)
# self.image = cv2.resize(self.image, (1024, 1024)) # self.image = cv2.resize(self.image, (1024, 1024))
images = [self.image.astype(np.uint8)] * self.batch_size images = [self.image.astype(np.uint8)] * self.batch_size