在 AI 绘画普及的今天，很多创作者都遇到过这样的困扰：需要批量处理上百张参考图生成新图，却只能一张张手动上传；生成的高清图片带着烦人的水印，还要额外找工具去除；Chrome 浏览器和驱动版本不兼容，折腾半天也无法启动；处理到一半程序崩溃，之前的工作全部白费。

作为一名每天需要处理大量 AI 图片的创作者，我也曾被这些问题折磨得苦不堪言，直到我开发了这款梦梦 AI 生图工具，彻底解决了这些痛点。

🔧 技术架构与核心原理

这款工具基于 Python+Selenium 技术栈开发，采用模块化设计思想，将 UI 交互层、浏览器控制层和业务逻辑层完全分离，确保了代码的可维护性和扩展性。界面使用原生 Tkinter 开发，无需安装额外的运行库，打包后即可直接运行。

为了提升用户体验，我专门设计了自定义样式按钮组件，支持悬停、点击等多种状态的视觉反馈：

class StyledButton(tk.Button):
    def __init__(self, master=None, **kwargs):
        self.bg_normal = kwargs.pop('bg', '#4CAF50')
        self.bg_hover = kwargs.pop('bg_hover', '#45a049')
        super().__init__(master, bg=self.bg_normal, **kwargs)
        self.bind('<Enter>', self.on_enter)
        self.bind('<Leave>', self.on_leave)
    
    def on_enter(self, event):
        if self['state'] != tk.DISABLED:
            self.config(bg=self.bg_hover)

这种设计不仅让界面更加美观，也避免了第三方 UI 库带来的体积膨胀问题，整个程序打包后体积不到 50MB。

🖼️ 批量图生图：告别重复劳动

批量处理是这款工具的核心功能，也是我开发它的初衷。传统的 AI 绘画工具一次只能处理一张图片，处理 100 张图片需要重复上传、输入提示词、点击生成等操作上百次，耗时又费力。

梦梦 AI 生图工具支持双目录批量上传模式，你可以同时设置参考图目录和循环图目录，工具会自动递归遍历所有子文件夹，按文件名中的数字智能排序：

def get_images_in_folder(self, folder_path):
    image_files = []
    valid_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp')
    for file in os.listdir(folder_path):
        file_path = os.path.join(folder_path, file)
        if os.path.isfile(file_path) and file.lower().endswith(valid_extensions):
            image_files.append(file_path)
    
    # 按文件名中的数字升序排序
    def extract_number(filepath):
        filename = os.path.basename(filepath)
        match = re.match(r'^(\d+)', filename)
        return int(match.group(1)) if match else float('inf')
    
    image_files.sort(key=extract_number)
    return image_files

你可以自由设置每个批次的图片数量、循环次数、生成尺寸（支持高清 2K 和超清 4K），以及全局固定后缀提示词。工具会自动按照你的设置批量生成图片，全程无需人工干预。

🧹 自动去水印：获取高清原图

很多 AI 绘画平台生成的图片都会带有水印，不仅影响美观，后期去除也非常麻烦。梦梦 AI 生图工具通过注入 JavaScript 代码的方式，在图片加载前就拦截了平台的 JSON 响应，直接获取原始无水印的高清图片 URL：

(function(){
    'use strict';
    let _parse=JSON.parse;
    JSON.parse=function(data){
        let jsonData=_parse(data);
        if(!data.match('creations'))return jsonData;
        let creations=findAllKeysInJson(jsonData,'creations');
        if(creations.length>0){
            creations.forEach((creation)=>{
                creation.map((item)=>{
                    const rawUrl=item.image.image_ori_raw.url;
                    item.image.image_ori.url=rawUrl;
                    return item;
                });
            })
        }
        return jsonData;
    };
})();

这种方式比后期使用 PS 或其他去水印工具处理效果更好，速度更快，而且完全不会损失图片质量。

⚙️ 智能驱动管理：零配置开箱即用

ChromeDriver 版本不兼容是使用 Selenium 开发的工具最常见的问题。为了解决这个问题，我在工具中集成了智能驱动管理功能，启动时会自动检测本地安装的 Chrome 浏览器版本，然后从多个镜像源自动下载匹配的 ChromeDriver 驱动：

def get_chrome_version_enhanced(self):
    try:
        reg_paths = [r"Software\Google\Chrome\BLBeacon", r"SOFTWARE\Google\Chrome\BLBeacon"]
        for reg_path in reg_paths:
            key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_READ)
            version, _ = winreg.QueryValueEx(key, "version")
            winreg.CloseKey(key)
            if version:
                return version
    except:
        pass
    return None

如果本地没有安装 Chrome 浏览器，工具还会自动下载并安装最新版本的 Chrome，真正实现了零配置开箱即用。

🔄 断点续传与错误恢复：稳定可靠

批量处理大量图片时，最担心的就是程序崩溃或网络中断导致之前的工作全部白费。梦梦 AI 生图工具内置了完善的断点续传机制，每处理完一个批次都会自动保存进度：

python

运行

def save_error_log(self):
    try:
        with open(self.ERROR_LOG_FILE, "w", encoding="utf-8") as f:
            f.write(f"{self.resume_folder_index},{self.resume_batch_index},{self.resume_image_index}")
    except Exception as e:
        self.update_status(f"保存错误记录失败: {str(e)}")

如果程序意外退出，下次启动时会自动检测未完成的任务，并从上次中断的地方继续处理。对于失败的批次，工具会自动重试最多 3 次，大大提高了批量处理的成功率。

📁 智能文件管理：自动分类命名

生成的图片管理也是一个让人头疼的问题。梦梦 AI 生图工具会自动按照原文件夹结构保存生成的图片，并根据参考图的文件名智能命名，避免了文件重名和混乱。你还可以自定义命名规则，满足不同的需求。

经过几个月的实际使用和不断优化，这款工具已经非常稳定可靠。我用它处理了上万张图片，效率比手动操作提升了 10 倍以上。如果你也需要批量处理 AI 图片，相信这款工具会成为你的得力助手。

📁 程序源码及成品软件下载：

https://pan.quark.cn/s/41d2eb656998

https://pan.baidu.com/s/5mrcU8ZvH71K_A4ljyy5aXA

---

AI 批量图生图工具，AI 图片批量生成器，剪映即梦批量生图，AI 去水印工具，自动 chromedriver 驱动更新，AI 图片批量处理，图生图批量工具，高清 AI 图片生成，AI 生图断点续传，智能图片命名