40 lines
979 B
Python
40 lines
979 B
Python
from PIL import Image
|
|
|
|
|
|
def to_standard_trimap(alpha, trimap):
|
|
# Alpha图生成 trimap
|
|
print(alpha)
|
|
image = Image.open(alpha)
|
|
print(image)
|
|
# image = image.convert("P")
|
|
# image_file.save('meinv_resize_trimap.png')
|
|
sp = image.size
|
|
width = sp[0]
|
|
height = sp[1]
|
|
|
|
for yh in range(height):
|
|
for xw in range(width):
|
|
dot = (xw, yh)
|
|
color_d_arr = image.getpixel(dot)
|
|
color_d=color_d_arr[0]
|
|
|
|
if 0 < color_d <= 60:
|
|
image.putpixel(dot, (0,0,0))
|
|
if 60 < color_d <= 200:
|
|
image.putpixel(dot, (128,128,128))
|
|
if 200 < color_d <= 255:
|
|
image.putpixel(dot, (255,255,255))
|
|
|
|
image.save(trimap)
|
|
|
|
|
|
|
|
|
|
|
|
# to_standard_trimap("..\\img\\trimap\\meinv_resize_trimap.png", "meinv_resize_bz_trimap.png")
|
|
#
|
|
#
|
|
# image = Image.open("meinv_resize_bz_trimap.png")
|
|
# data = image.getdata()
|
|
# np.savetxt("data4.txt", data,fmt='%d',delimiter=',')
|