feat(fallback image): use extracted assets to get image

This commit is contained in:
Haoyu Xu
2021-08-02 17:28:55 -04:00
parent 39e492d1bf
commit 8d9efe8046
4 changed files with 34 additions and 11 deletions

View File

@@ -22,6 +22,9 @@ class AlphaComposite:
def __open_image(self):
self.image = Image.open(self.image_path)
self.mask = Image.open(self.mask_path)
if self.image.size != self.mask.size:
# resize mask
self.mask = self.__resize(self.mask, self.image.size[0], self.image.size[1])
self.loaded_image = self.image.load()
self.loaded_mask = self.mask.load()
@@ -29,4 +32,8 @@ class AlphaComposite:
for y in range(self.mask.size[1]):
for x in range(self.mask.size[0]):
self.loaded_image[x, y] = (self.loaded_image[x, y][0], self.loaded_image[x, y][1], self.loaded_image[x, y][2], self.loaded_mask[x, y][2])
self.image.save(pathlib.Path.cwd().joinpath(self.save_path))
self.image.save(pathlib.Path.cwd().joinpath(self.save_path))
# resize image
def __resize(self, image, width: int, height: int):
return image.resize((width, height))