002.01 图片移除背景成PNG文件

import ctypes
from io import BytesIO
from pathlib import Path
from PIL import Image
import PySimpleGUI as sgCopy

class Picture():

    def __init__(self):
        self.im = None # 处理中的图片
        self.old_im = None # 保留图片供 Undo
        self.default_tolerance = 16 # 预设颜色偏差值
        self.tolerances = [0, 8, 16, 32, 64, 128] # 可选颜色偏差值
        self.picture = None # 显示的图片对象
        self.width, self.height = self.size = (1600, 800) # 图片显示区尺寸
        self.transparency = (0, 0, 0, 0) # 用来取代的透明值Copy

    def limit(self, color, tolerance):
        result = []
        for c in color:
            result += [
                min(max(c-tolerance, 0), 255), min(max(c+tolerance, 0), 255)]
        return resultCopy

    def change(self, x, y):
        r, g, b, a = self.im.getpixel((x, y))
        e = self.tolerance
        self.r0, self.r1, self.g0, self.g1, self.b0, self.b1 = self.limit(
            (r, g, b), self.tolerance)
        self.old_im = self.im.copy()
        self.update(x, y)Copy

    def is_same(self, x, y):
        r, g, b, a = self.im.getpixel((x, y))
        return True if (
            self.r0 <= r <= self.r1 and
            self.g0 <= g <= self.g1 and
            self.b0 <= b <= self.b1 and
            a != 0
            ) else FalseCopy

    def update(self, x, y):
        checked = [[False for y in range(self.im.height)]
            for x in range(self.im.width)]
        to_do = [(x, y)]
        while to_do:
            temp = []
            for x, y in to_do:
                checked[x][y] = True
                if self.is_same(x, y):
                    self.im.putpixel((x, y), self.transparency)
                    if x-1 >= 0 and not checked[x-1][y]:
                        temp.append((x-1, y))
                    if x+1 < self.im.width and not checked[x+1][y]:
                        temp.append((x+1, y))
                    if y-1 >= 0 and not checked[x][y-1]:
                        temp.append((x, y-1))
                    if y+1 < self.im.height and not checked[x][y+1]:
                        temp.append((x, y+1))
            to_do = temp.copy()Copy

    def new(self):
        if self.picture:
            graph.delete_figure(self.picture)
        with BytesIO() as output:
            self.im.save(output, format="PNG")
            data = output.getvalue()
        self.picture = graph.draw_image(data=data, location=(self.x0, self.y0))Copy

    def click(self, position):
        if self.picture is None:
            return
        x1, y1 = position
        x, y = x1-self.x0, self.y0-y1
        if not (0<=x<self.im.width and 0<=y<self.im.height):
            return
        self.change(x, y)
        self.new()Copy

    def open(self):
        filename = sg.popup_get_file('Open new image', no_window=True, modal=True)
        if filename and Path(filename).is_file():
            try:
                self.im = Image.open(filename).convert(mode='RGBA')
            except:
                return
            self.x0 = (self.width-self.im.width)//2
            self.y0 = (self.height+self.im.height)//2
            self.tolerance = self.default_tolerance
            window['TOLERANCE'].update(value=self.tolerance)
            self.new()Copy

    def save(self):
        if self.im is None:
            return
        filename = sg.popup_get_file('Save new image', no_window=True,
            modal=True, save_as=True)
        if filename:
            if not filename.lower().endswith(".png"):
                filename += ".png"
            self.im.save(filename)
        returnCopy

    def undo(self):
        if self.old_im:
            self.im = self.old_im.copy()
            self.new()Copy

ctypes.windll.user32.SetProcessDPIAware() # Set unit of GUI to pixels
font = ('Courier New', 16, 'bold')
p = Picture()Copy

layout = [
    [sg.Button("OPEN", font=font),
     sg.Button("SAVE", font=font),
     sg.Button("UNDO", font=font),
     sg.Text("Tolerance", font=font),
     sg.Combo(p.tolerances, p.default_tolerance, font=font, key='TOLERANCE',
        enable_events=True)],
    [sg.Graph(p.size, (0, 0), p.size, enable_events=True, key='GRAPH',
        background_color='darkgreen')],
]Copy

window = sg.Window('Move Outline of Image', layout, finalize=True)
graph = window['GRAPH']Copy

while True:

    event, values = window.read()

    if event == sg.WINDOW_CLOSED:
        break
    elif event == 'OPEN':
        p.open()
    elif event == 'SAVE':
        p.save()
    elif event == 'UNDO':
        p.undo()
    elif event == 'TOLERANCE':
        p.tolerance = values['TOLERANCE']
    elif event == 'GRAPH':
        p.click(values['GRAPH'])Copy

window.close()Copy