feat(新功能):

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试): 旧版product 测试
This commit is contained in:
zhouchengrong
2025-02-03 19:13:48 +08:00
parent e1c00bbd66
commit 74008c4586

View File

@@ -263,7 +263,8 @@ class GenerateProductImage:
def get_result(self):
try:
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))
images = [self.image.astype(np.uint8)] * self.batch_size