python png白色背景转透明背景

个人博客，已转移至http://www.zgeo.work:8022/wordpress

python png白色背景转透明背景,复制文件后可直接运行

# -*- coding: utf-8 -*-
import os
from PIL import Image

def transparent_back(filename):
    if(filename.find("png")>=0):
        img = Image.open(filename)
        L, H = img.size
        color_0 = (255,255,255,255)#要替换的颜色
        is_saved=False
        for h in range(H):
            for l in range(L):
                dot = (l,h)
                color_1 = img.getpixel(dot)
                if color_1 == color_0:
                    is_saved=True
                    color_1 = color_1[:-1] + (0,)
                    img.putpixel(dot,color_1)
        if is_saved :
            img.save(filename, "PNG")
               
def get_files_name(current_dir):
    for file_list in os.walk(current_dir):
         filesubdir=file_list[0]
         filenames=file_list[2]
         print("current_dir--"+ file_list[0])
         if(len(filenames)>0):
            for i in filenames:
                transparent_back(filesubdir+"\\"+i)

get_files_name(png_dir)