fixed wrong image format conversion.

This commit is contained in:
bruefire
2024-11-11 04:10:41 +09:00
parent b2cac6f6a9
commit 0284a7e3eb
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -93,7 +93,7 @@ class LoadImagesInpaint:
s = ImageOps.exif_transpose(s).convert("RGBA")
s = s.resize(i.size)
image = Image.alpha_composite(i, s)
image.convert("RGB")
image = image.convert("RGB")
image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,]