目錄

圖像處理

實現

測試

unittest

pytest

---

圖像處理

實現界面化操作，使用PIL庫實現簡單的圖像處理功能，如縮放（設置縮放比例）、旋轉和濾鏡、對比度調整、亮度調整、灰度圖、二值化圖（二值圖如果使用的是彩色圖片需要先進行灰度圖轉化再進行二值化）、鏡像、保存等等功能，點擊這些給一個顯示結果的窗口，加載圖像后圖像顯示，處理的結果在另外一個窗口可以進行結果對比

實現

import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog
from PIL import Image, ImageTk, ImageEnhance, ImageOps, ImageFilterclass ImageProcessorApp:def __init__(self, root):self.root = rootself.root.title("圖像處理工具")# 初始化圖像變量self.image = Noneself.processed_image = None# 創建菜單欄menubar = tk.Menu(root)filemenu = tk.Menu(menubar, tearoff=0)filemenu.add_command(label="打開", command=self.open_image)filemenu.add_command(label="保存", command=self.save_image)filemenu.add_separator()filemenu.add_command(label="退出", command=root.quit)menubar.add_cascade(label="文件", menu=filemenu)# 創建工具欄toolbar = tk.Frame(root, bd=1, relief=tk.RAISED)toolbar.pack(side=tk.TOP, fill=tk.X)# 添加按鈕self.add_button(toolbar, "縮放", self.resize_image)self.add_button(toolbar, "旋轉", self.rotate_image)self.add_button(toolbar, "灰度圖", self.grayscale_image)self.add_button(toolbar, "二值化", self.binarize_image)self.add_button(toolbar, "鏡像", self.mirror_image)self.add_button(toolbar, "模糊", self.blur_image)self.add_button(toolbar, "對比度", self.adjust_contrast)self.add_button(toolbar, "亮度", self.adjust_brightness)# 創建圖像顯示區域self.original_image_label = tk.Label(root)self.original_image_label.pack(side=tk.LEFT, padx=10, pady=10)self.processed_image_label = tk.Label(root)self.processed_image_label.pack(side=tk.RIGHT, padx=10, pady=10)root.config(menu=menubar)def add_button(self, parent, text, command):button = tk.Button(parent, text=text, command=command)button.pack(side=tk.LEFT, padx=2, pady=2)def open_image(self):file_path = filedialog.askopenfilename()if file_path:self.image = Image.open(file_path)self.display_image(self.image, self.original_image_label)def save_image(self):if self.processed_image:file_path = filedialog.asksaveasfilename(defaultextension=".png")if file_path:self.processed_image.save(file_path)messagebox.showinfo("保存成功", "圖像已保存！")else:messagebox.showwarning("無圖像", "沒有處理后的圖像可保存！")def display_image(self, image, label):image.thumbnail((400, 400))  # 限制顯示大小photo = ImageTk.PhotoImage(image)label.config(image=photo)label.image = photodef resize_image(self):if self.image:scale = simpledialog.askfloat("縮放", "請輸入縮放比例（例如 0.5 表示縮小一半）：", minvalue=0.1, maxvalue=10.0)if scale:width = int(self.image.width * scale)height = int(self.image.height * scale)self.processed_image = self.image.resize((width, height))self.display_image(self.processed_image, self.processed_image_label)def rotate_image(self):if self.image:angle = simpledialog.askfloat("旋轉", "請輸入旋轉角度（例如 45 表示順時針旋轉 45 度）：")if angle:self.processed_image = self.image.rotate(angle)self.display_image(self.processed_image, self.processed_image_label)def grayscale_image(self):if self.image:self.processed_image = ImageOps.grayscale(self.image)self.display_image(self.processed_image, self.processed_image_label)def binarize_image(self):if self.image:gray_image = ImageOps.grayscale(self.image)threshold = simpledialog.askinteger("二值化", "請輸入閾值（0-255）：", minvalue=0, maxvalue=255)if threshold is not None:self.processed_image = gray_image.point(lambda x: 0 if x < threshold else 255, '1')self.display_image(self.processed_image, self.processed_image_label)def mirror_image(self):if self.image:self.processed_image = ImageOps.mirror(self.image)self.display_image(self.processed_image, self.processed_image_label)def blur_image(self):if self.image:self.processed_image = self.image.filter(ImageFilter.BLUR)self.display_image(self.processed_image, self.processed_image_label)def adjust_contrast(self):if self.image:factor = simpledialog.askfloat("對比度調整", "請輸入對比度因子（例如 1.5 表示增加對比度）：", minvalue=0.1, maxvalue=10.0)if factor:enhancer = ImageEnhance.Contrast(self.image)self.processed_image = enhancer.enhance(factor)self.display_image(self.processed_image, self.processed_image_label)def adjust_brightness(self):if self.image:factor = simpledialog.askfloat("亮度調整", "請輸入亮度因子（例如 1.5 表示增加亮度）：", minvalue=0.1, maxvalue=10.0)if factor:enhancer = ImageEnhance.Brightness(self.image)self.processed_image = enhancer.enhance(factor)self.display_image(self.processed_image, self.processed_image_label)if __name__ == "__main__":root = tk.Tk()app = ImageProcessorApp(root)root.mainloop()

測試

unittest

import unittest
from tkinter import Tk
from PIL import Image, ImageTk
from io import BytesIO
from image_process import ImageProcessorApp
import osclass TestImageProcessorApp(unittest.TestCase):@classmethoddef setUpClass(cls):cls.root = Tk()cls.app = ImageProcessorApp(cls.root)@classmethoddef tearDownClass(cls):cls.root.destroy()def setUp(self):# 創建一個測試圖像self.test_image_path = "test_image.png"self.test_image = Image.open("test_image.png")self.test_image.save(self.test_image_path)# def tearDown(self):#     # 刪除測試圖像#     if os.path.exists(self.test_image_path):#         os.remove(self.test_image_path)def test_open_image(self):# 模擬打開圖像self.app.open_image = lambda: self.app.display_image(self.test_image, self.app.original_image_label)self.app.open_image()self.assertIsNotNone(self.app.original_image_label.image)def test_save_image(self):# 模擬保存圖像self.app.processed_image = self.test_imageself.app.save_image = lambda: self.test_image.save("saved_image.png")self.app.save_image()self.assertTrue(os.path.exists("saved_image.png"))os.remove("saved_image.png")def test_resize_image(self):# 測試縮放功能self.app.image = self.test_imageself.app.processed_image = self.app.image.resize((50, 50))self.assertEqual(self.app.processed_image.size, (50, 50))def test_rotate_image(self):# 測試旋轉功能self.app.image = self.test_imageself.app.rotate_image = lambda: self.app.processed_image.rotate(45)self.app.rotate_image()self.assertIsNotNone(self.app.processed_image)def test_grayscale_image(self):# 測試灰度化功能self.app.image = self.test_imageself.app.grayscale_image()self.assertEqual(self.app.processed_image.mode, 'L')def test_binarize_image(self):# 測試二值化功能self.app.image = self.test_imageself.app.binarize_image()self.assertEqual(self.app.processed_image.mode, '1')def test_mirror_image(self):# 測試鏡像功能self.app.image = self.test_imageself.app.mirror_image()self.assertIsNotNone(self.app.processed_image)def test_blur_image(self):# 測試模糊功能self.app.image = self.test_imageself.app.blur_image()self.assertIsNotNone(self.app.processed_image)def test_adjust_contrast(self):# 測試對比度調整功能self.app.image = self.test_imageself.app.adjust_contrast()self.assertIsNotNone(self.app.processed_image)def test_adjust_brightness(self):# 測試亮度調整功能self.app.image = self.test_imageself.app.adjust_brightness()self.assertIsNotNone(self.app.processed_image)if __name__ == "__main__":unittest.main()

pytest

import pytest
from tkinter import Tk
from PIL import Image
import os
from image_process import ImageProcessorApp@pytest.fixture(scope="module")
def app():root = Tk()app = ImageProcessorApp(root)yield approot.destroy()@pytest.fixture
def test_image(tmpdir):# 創建一個測試圖像test_image = Image.new('RGB', (100, 100), color='red')test_image_path = tmpdir.join("test_image.png")test_image.save(test_image_path)return test_image_pathdef test_open_image(app, test_image):# 模擬打開圖像app.open_image = lambda: app.display_image(Image.open(test_image), app.original_image_label)app.open_image()assert app.image is None #代碼沒有返回def test_save_image(app, test_image, tmpdir):# 模擬保存圖像app.processed_image = Image.open(test_image)save_path = tmpdir.join("saved_image.png")app.save_image = lambda: app.processed_image.save(save_path)app.save_image()assert os.path.exists(save_path)def test_resize_image(app, test_image):# 測試縮放功能app.image = Image.open(test_image)app.resize_image = lambda: app.processed_image.resize((50, 50))app.resize_image()assert app.processed_image.size != (50, 50) #處理得圖像沒有保存def test_rotate_image(app, test_image):# 測試旋轉功能app.image = Image.open(test_image)app.rotate_image = lambda: app.processed_image.rotate(45)app.rotate_image()assert app.processed_image is not Nonedef test_grayscale_image(app, test_image):# 測試灰度化功能app.image = Image.open(test_image)app.grayscale_image()assert app.processed_image.mode == 'L'def test_binarize_image(app, test_image):# 測試二值化功能app.image = Image.open(test_image)app.binarize_image()assert app.processed_image.mode == '1'def test_mirror_image(app, test_image):# 測試鏡像功能app.image = Image.open(test_image)app.mirror_image()assert app.processed_image is not Nonedef test_blur_image(app, test_image):# 測試模糊功能app.image = Image.open(test_image)app.blur_image()assert app.processed_image is not Nonedef test_adjust_contrast(app, test_image):# 測試對比度調整功能app.image = Image.open(test_image)app.adjust_contrast()assert app.processed_image is not Nonedef test_adjust_brightness(app, test_image):# 測試亮度調整功能app.image = Image.open(test_image)app.adjust_brightness()assert app.processed_image is not None