python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python图片压缩

Python中图片压缩小工具的开发与异常解决详解

作者:晨曦之光Wing

这篇文章主要为大家详细介绍了如何通过Deep Seek编写Python代码,制作一个图片压缩小工具,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下

一、问题描述

通过Deep Seek写Python代码,制作一个图片压缩小工具,结果发现竖图压缩失败,然后一直让它优化,修复这个问题,结果还是一直失败。后面一步步调试才发现,根本不是横图、竖图的问题,而是某些特定场景(图片大小、压缩质量)下会导致死循环,进而导致压缩失败。

二、问题代码及修复

下方while循环的>=会导致某些特定场景下出现死循环,将等号去掉即可。

    def compress_image(self, input_path, output_path, target_size_kb):
        """压缩单张图片到目标大小"""
        # 打开图片
        img = Image.open(input_path)

        # 如果是PNG格式,转换为JPG以获得更好的压缩效果
        if img.format == 'PNG':
            img = img.convert('RGB')
            output_path = output_path.rsplit('.', 1)[0] + '.jpg'

        # 设置初始质量
        quality = 95
        min_quality = 10

        # 转换为字节并检查大小
        while quality >= min_quality:
            # 保存为字节数据以检查大小
            img.save(output_path, quality=quality, optimize=True)

            # 检查文件大小
            file_size_kb = os.path.getsize(output_path) / 1024

            if file_size_kb <= target_size_kb:
                break

            # 质量降低步长根据差距动态调整
            size_ratio = file_size_kb / target_size_kb
            quality_reduction = max(5, int(quality * (1 - 1 / size_ratio) / 2))
            quality = max(min_quality, quality - quality_reduction)

三、修复历程

经过上述几个来回,发现问题依然没有解决,于是我就一步步调试代码去了,此时我以为这个问题是一步步“优化”出来的。

【第一个版本】

后面看了一下发现,前面七个版本基本都是类似的处理,估计我是在第四个版本的时候使用了那张竖图,导致压缩失败了,然后以“失败的结论”去修复失败的问题,结果就一直失败。开了深度思考之后,又优化了两个版本,下面是最后一个版本。

【最后一个版本】

给我整笑了

成功了?我图呢 ?_?

此时我已经不想去纠结了……反正第一个版本修改之后能正常使用了

四、正确代码

第一个版本修改之后的代码:

import os
import sys
import tkinter as tk
from tkinter import filedialog, messagebox, ttk
from PIL import Image
import threading


class ImageCompressorApp:
    def __init__(self, root):
        self.root = root
        self.root.title("图片压缩工具")
        self.root.geometry("500x300")
        self.root.resizable(False, False)

        # 变量初始化
        self.input_folder = tk.StringVar()
        self.output_folder = tk.StringVar()
        self.target_size = tk.IntVar(value=100)  # 默认100KB

        self.create_widgets()

    def create_widgets(self):
        # 输入文件夹选择
        tk.Label(self.root, text="输入文件夹:").grid(row=0, column=0, padx=10, pady=10, sticky="w")
        tk.Entry(self.root, textvariable=self.input_folder, width=40).grid(row=0, column=1, padx=10, pady=10)
        tk.Button(self.root, text="浏览", command=self.browse_input).grid(row=0, column=2, padx=10, pady=10)

        # 输出文件夹选择
        tk.Label(self.root, text="输出文件夹:").grid(row=1, column=0, padx=10, pady=10, sticky="w")
        tk.Entry(self.root, textvariable=self.output_folder, width=40).grid(row=1, column=1, padx=10, pady=10)
        tk.Button(self.root, text="浏览", command=self.browse_output).grid(row=1, column=2, padx=10, pady=10)

        # 目标大小设置
        tk.Label(self.root, text="目标大小(KB):").grid(row=2, column=0, padx=10, pady=10, sticky="w")
        tk.Scale(self.root, from_=10, to=1000, orient=tk.HORIZONTAL, variable=self.target_size).grid(row=2, column=1,
                                                                                                     padx=10, pady=10,
                                                                                                     sticky="ew")

        # 压缩按钮
        self.compress_btn = tk.Button(self.root, text="开始压缩", command=self.start_compression, bg="lightblue")
        self.compress_btn.grid(row=3, column=1, padx=10, pady=20)

        # 进度条
        self.progress = ttk.Progressbar(self.root, orient=tk.HORIZONTAL, length=400, mode='indeterminate')
        self.progress.grid(row=4, column=0, columnspan=3, padx=10, pady=10)

        # 状态标签
        self.status_label = tk.Label(self.root, text="准备就绪", relief=tk.SUNKEN, anchor=tk.W)
        self.status_label.grid(row=5, column=0, columnspan=3, sticky="ew", padx=10, pady=10)

    def browse_input(self):
        folder = filedialog.askdirectory()
        if folder:
            self.input_folder.set(folder)

    def browse_output(self):
        folder = filedialog.askdirectory()
        if folder:
            self.output_folder.set(folder)

    def start_compression(self):
        if not self.input_folder.get() or not self.output_folder.get():
            messagebox.showerror("错误", "请选择输入和输出文件夹")
            return

        # 在后台线程中执行压缩,避免界面冻结
        thread = threading.Thread(target=self.compress_images)
        thread.daemon = True
        thread.start()

    def compress_images(self):
        self.compress_btn.config(state=tk.DISABLED)
        self.progress.start()
        self.status_label.config(text="正在压缩图片...")

        input_dir = self.input_folder.get()
        output_dir = self.output_folder.get()
        target_size_kb = self.target_size.get()

        # 支持的图片格式
        supported_formats = ('.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp')

        try:
            # 获取所有图片文件
            image_files = [f for f in os.listdir(input_dir)
                           if f.lower().endswith(supported_formats)]

            if not image_files:
                messagebox.showinfo("信息", "未找到支持的图片文件")
                return

            # 创建输出目录(如果不存在)
            if not os.path.exists(output_dir):
                os.makedirs(output_dir)

            # 处理每张图片
            for i, filename in enumerate(image_files):
                self.status_label.config(text=f"正在处理 {i + 1}/{len(image_files)}: {filename}")
                self.root.update()

                input_path = os.path.join(input_dir, filename)
                output_path = os.path.join(output_dir, filename)

                # 压缩图片
                self.compress_image(input_path, output_path, target_size_kb)

            self.status_label.config(text=f"完成! 已压缩 {len(image_files)} 张图片")
            messagebox.showinfo("完成", f"图片压缩完成! 共处理 {len(image_files)} 张图片")

        except Exception as e:
            messagebox.showerror("错误", f"压缩过程中发生错误: {str(e)}")
        finally:
            self.progress.stop()
            self.compress_btn.config(state=tk.NORMAL)

    def compress_image(self, input_path, output_path, target_size_kb):
        """压缩单张图片到目标大小"""
        # 打开图片
        img = Image.open(input_path)

        # 如果是PNG格式,转换为JPG以获得更好的压缩效果
        if img.format == 'PNG':
            img = img.convert('RGB')
            output_path = output_path.rsplit('.', 1)[0] + '.jpg'

        # 设置初始质量
        quality = 95
        min_quality = 10

        # 转换为字节并检查大小
        while quality > min_quality:
            # 保存为字节数据以检查大小
            img.save(output_path, quality=quality, optimize=True)

            # 检查文件大小
            file_size_kb = os.path.getsize(output_path) / 1024

            if file_size_kb <= target_size_kb:
                break

            # 质量降低步长根据差距动态调整
            size_ratio = file_size_kb / target_size_kb
            quality_reduction = max(5, int(quality * (1 - 1 / size_ratio) / 2))
            quality = max(min_quality, quality - quality_reduction)


def main():
    root = tk.Tk()
    app = ImageCompressorApp(root)
    root.mainloop()


if __name__ == "__main__":
    main()

原图998KB,压缩后140KB,压缩格式为jpg,支持批量压缩(压缩所选文件夹下的所有图片)

(P.S. 后续将发布进一步优化的版本)

到此这篇关于Python中图片压缩小工具的开发与异常解决详解的文章就介绍到这了,更多相关Python图片压缩内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文