feat(新功能):
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试): 旧版product 测试
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user