Spaces:
Sleeping
Sleeping
Update render/canvas.py
Browse files- render/canvas.py +3 -3
render/canvas.py
CHANGED
|
@@ -21,11 +21,11 @@ class Canvas:
|
|
| 21 |
self.img.putpixel(dot, color + (255,))
|
| 22 |
|
| 23 |
def add_white_border(self, border_size=5):
|
| 24 |
-
|
| 25 |
if self.img.mode != "RGBA":
|
| 26 |
self.img = self.img.convert("RGBA")
|
| 27 |
|
| 28 |
-
|
| 29 |
alpha = self.img.getchannel("A")
|
| 30 |
# print(alpha.size)
|
| 31 |
dilated_alpha = alpha.filter(ImageFilter.MaxFilter(size=5))
|
|
@@ -33,7 +33,7 @@ class Canvas:
|
|
| 33 |
white_area = Image.new("RGBA", self.img.size, (255, 255, 255, 255))
|
| 34 |
white_area.putalpha(dilated_alpha)
|
| 35 |
|
| 36 |
-
|
| 37 |
result = Image.alpha_composite(white_area, self.img)
|
| 38 |
# expanded_alpha = ImageOps.expand(alpha, border=border_size, fill=255)
|
| 39 |
# white_border = Image.new("RGBA", image.size, (255, 255, 255, 255))
|
|
|
|
| 21 |
self.img.putpixel(dot, color + (255,))
|
| 22 |
|
| 23 |
def add_white_border(self, border_size=5):
|
| 24 |
+
|
| 25 |
if self.img.mode != "RGBA":
|
| 26 |
self.img = self.img.convert("RGBA")
|
| 27 |
|
| 28 |
+
|
| 29 |
alpha = self.img.getchannel("A")
|
| 30 |
# print(alpha.size)
|
| 31 |
dilated_alpha = alpha.filter(ImageFilter.MaxFilter(size=5))
|
|
|
|
| 33 |
white_area = Image.new("RGBA", self.img.size, (255, 255, 255, 255))
|
| 34 |
white_area.putalpha(dilated_alpha)
|
| 35 |
|
| 36 |
+
|
| 37 |
result = Image.alpha_composite(white_area, self.img)
|
| 38 |
# expanded_alpha = ImageOps.expand(alpha, border=border_size, fill=255)
|
| 39 |
# white_border = Image.new("RGBA", image.size, (255, 255, 255, 255))
|