38 lines
848 B
Python
38 lines
848 B
Python
from PIL import Image
|
|
import numpy as np
|
|
|
|
|
|
|
|
def alpha_cutout(img, threshold=100, dist=5):
|
|
arr = np.array(np.asarray(img)) # 获取图像数据，使用了numpy
|
|
r, g, b, a = np.rollaxis(arr, axis=-1)
|
|
|
|
mask = ((r > threshold)
|
|
& (g > threshold)
|
|
& (b > threshold)
|
|
& (np.abs(r - g) < dist) # 将接近白色背景的也替换掉
|
|
& (np.abs(r - b) < dist)
|
|
& (np.abs(g - b) < dist)
|
|
)
|
|
arr[mask, 3] = 0
|
|
|
|
img = Image.fromarray(arr, mode='RGBA') # 转换为图像格式
|
|
return img
|
|
|
|
def cutout_csdn():
|
|
|
|
threshold = 100
|
|
dist = 5
|
|
img = Image.open("img.png").convert('RGBA') # 增加Alpha通道
|
|
|
|
img = alpha_cutout(img, threshold, dist)
|
|
|
|
|
|
img.show()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cutout_csdn()
|
|
# image = "img.png"
|
|
# do_cutout(image)
|