快捷導(dǎo)航

使用Python和Pillow實現(xiàn)圖片馬賽克功能

更新時間：2024年08月15日 10:27:31 作者：winfredzhang

在這篇博客中,我們將探討如何使用Python創(chuàng)建一個簡單而有趣的桌面應(yīng)用程序,我們的目標(biāo)是構(gòu)建一個應(yīng)用,允許用戶選擇一張照片,然后在照片的右下角添加馬賽克效果,感興趣的小伙伴跟著小編一起來看看吧

引言

在這篇博客中,我們將探討如何使用Python創(chuàng)建一個簡單而有趣的桌面應(yīng)用程序。我們的目標(biāo)是構(gòu)建一個應(yīng)用,允許用戶選擇一張照片,然后在照片的右下角添加馬賽克效果。這個項目將展示如何結(jié)合使用wxPython來創(chuàng)建圖形用戶界面(GUI)和Pillow庫來處理圖像。

項目概述

我們的應(yīng)用程序?qū)⒕哂幸韵鹿δ?

一個用于選擇照片文件的按鈕
一個用于應(yīng)用馬賽克效果的按鈕
一個顯示當(dāng)前狀態(tài)的文本標(biāo)簽

當(dāng)用戶點擊"添加馬賽克"按鈕時,程序?qū)⒃谶x定照片的右下角添加馬賽克效果,并將結(jié)果保存為新的圖片文件。

所需庫

為了構(gòu)建這個應(yīng)用,我們需要安裝兩個主要的庫:

wxPython: 用于創(chuàng)建圖形用戶界面
Pillow: 用于圖像處理

你可以使用pip安裝這些庫:

pip install wxPython Pillow

全部代碼

import wx
import os
from PIL import Image

class PhotoMosaicApp(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='Photo Mosaic App')
        panel = wx.Panel(self)
        
        self.photo_path = None
        
        # 創(chuàng)建控件
        self.select_button = wx.Button(panel, label='選擇照片')
        self.mosaic_button = wx.Button(panel, label='添加馬賽克')
        self.status_text = wx.StaticText(panel, label='請選擇一張照片')
        
        # 綁定事件
        self.select_button.Bind(wx.EVT_BUTTON, self.on_select)
        self.mosaic_button.Bind(wx.EVT_BUTTON, self.on_mosaic)
        
        # 設(shè)置布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.select_button, 0, wx.ALL | wx.CENTER, 5)
        sizer.Add(self.mosaic_button, 0, wx.ALL | wx.CENTER, 5)
        sizer.Add(self.status_text, 0, wx.ALL | wx.CENTER, 5)
        
        panel.SetSizer(sizer)
        self.Show()
    
    def on_select(self, event):
        with wx.FileDialog(self, "選擇照片", wildcard="圖片文件 (*.jpg;*.png)|*.jpg;*.png",
                           style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
            if fileDialog.ShowModal() == wx.ID_CANCEL:
                return
            self.photo_path = fileDialog.GetPath()
            self.status_text.SetLabel(f'已選擇: {os.path.basename(self.photo_path)}')
    
    def on_mosaic(self, event):
        if not self.photo_path:
            wx.MessageBox('請先選擇一張照片', '提示', wx.OK | wx.ICON_INFORMATION)
            return
        
        try:
            # 打開圖片
            img = Image.open(self.photo_path)
            width, height = img.size
            
            # 定義右下角區(qū)域
            mosaic_size = min(width, height) // 4
            mosaic_area = (width - mosaic_size, height - mosaic_size, width, height)
            
            # 裁剪右下角區(qū)域并縮小
            mosaic = img.crop(mosaic_area).resize((20, 20), Image.NEAREST)
            
            # 將縮小的區(qū)域放大回原始大小，形成馬賽克效果
            mosaic = mosaic.resize((mosaic_size, mosaic_size), Image.NEAREST)
            
            # 將馬賽克區(qū)域粘貼回原圖
            img.paste(mosaic, (width - mosaic_size, height - mosaic_size))
            
            # 保存結(jié)果
            output_path = os.path.splitext(self.photo_path)[0] + '_mosaic.png'
            img.save(output_path)
            
            self.status_text.SetLabel(f'馬賽克添加成功，已保存為: {os.path.basename(output_path)}')
        except Exception as e:
            wx.MessageBox(f'處理圖片時出錯: {str(e)}', '錯誤', wx.OK | wx.ICON_ERROR)

if __name__ == '__main__':
    app = wx.App()
    frame = PhotoMosaicApp()
    app.MainLoop()

代碼實現(xiàn)

讓我們逐步實現(xiàn)這個應(yīng)用程序:

首先,我們導(dǎo)入必要的模塊并創(chuàng)建主應(yīng)用類:

import wx
import os
from PIL import Image

class PhotoMosaicApp(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='Photo Mosaic App')
        panel = wx.Panel(self)
        
        self.photo_path = None
        
        # 創(chuàng)建控件
        self.select_button = wx.Button(panel, label='選擇照片')
        self.mosaic_button = wx.Button(panel, label='添加馬賽克')
        self.status_text = wx.StaticText(panel, label='請選擇一張照片')
        
        # 綁定事件
        self.select_button.Bind(wx.EVT_BUTTON, self.on_select)
        self.mosaic_button.Bind(wx.EVT_BUTTON, self.on_mosaic)
        
        # 設(shè)置布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.select_button, 0, wx.ALL | wx.CENTER, 5)
        sizer.Add(self.mosaic_button, 0, wx.ALL | wx.CENTER, 5)
        sizer.Add(self.status_text, 0, wx.ALL | wx.CENTER, 5)
        
        panel.SetSizer(sizer)
        self.Show()

接下來,我們實現(xiàn)選擇照片的功能:

def on_select(self, event):
    with wx.FileDialog(self, "選擇照片", wildcard="圖片文件 (*.jpg;*.png)|*.jpg;*.png",
                       style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
        if fileDialog.ShowModal() == wx.ID_CANCEL:
            return
        self.photo_path = fileDialog.GetPath()
        self.status_text.SetLabel(f'已選擇: {os.path.basename(self.photo_path)}')

然后,我們實現(xiàn)添加馬賽克的核心功能:

def on_mosaic(self, event):
    if not self.photo_path:
        wx.MessageBox('請先選擇一張照片', '提示', wx.OK | wx.ICON_INFORMATION)
        return
    
    try:
        # 打開圖片
        img = Image.open(self.photo_path)
        width, height = img.size
        
        # 定義右下角區(qū)域
        mosaic_size = min(width, height) // 4
        mosaic_area = (width - mosaic_size, height - mosaic_size, width, height)
        
        # 裁剪右下角區(qū)域并縮小
        mosaic = img.crop(mosaic_area).resize((20, 20), Image.NEAREST)
        
        # 將縮小的區(qū)域放大回原始大小,形成馬賽克效果
        mosaic = mosaic.resize((mosaic_size, mosaic_size), Image.NEAREST)
        
        # 將馬賽克區(qū)域粘貼回原圖
        img.paste(mosaic, (width - mosaic_size, height - mosaic_size))
        
        # 保存結(jié)果
        output_path = os.path.splitext(self.photo_path)[0] + '_mosaic.png'
        img.save(output_path)
        
        self.status_text.SetLabel(f'馬賽克添加成功,已保存為: {os.path.basename(output_path)}')
    except Exception as e:
        wx.MessageBox(f'處理圖片時出錯: {str(e)}', '錯誤', wx.OK | wx.ICON_ERROR)

最后,我們添加主程序入口:

if __name__ == '__main__':
    app = wx.App()
    frame = PhotoMosaicApp()
    app.MainLoop()

代碼解析

讓我們來解析一下關(guān)鍵部分的代碼:

GUI創(chuàng)建: 我們使用wxPython創(chuàng)建了一個簡單的窗口,包含兩個按鈕和一個狀態(tài)文本。wx.BoxSizer用于垂直排列這些控件。
文件選擇: wx.FileDialog用于創(chuàng)建一個文件選擇對話框,允許用戶選擇JPG或PNG格式的圖片。
圖像處理:
- 我們使用Pillow的Image.open()方法打開選定的圖片。
- 通過計算圖片尺寸的1/4來確定馬賽克區(qū)域的大小。
- 使用crop()方法裁剪右下角區(qū)域。
- 將裁剪區(qū)域縮小到20x20像素,然后再放大回原始大小,創(chuàng)造馬賽克效果。
- 最后使用paste()方法將馬賽克區(qū)域粘貼回原圖。
異常處理: 我們使用try-except塊來捕獲和處理可能發(fā)生的錯誤,提高應(yīng)用程序的健壯性。