MohmedAnik commited on
Commit
6522e8a
·
verified ·
1 Parent(s): 67ae3bc

Update render/canvas.py

Browse files
Files changed (1) hide show
  1. 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
- # 确保输入图像是 RGBA 模式
25
  if self.img.mode != "RGBA":
26
  self.img = self.img.convert("RGBA")
27
 
28
- # 提取 alpha 通道
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))