【应用程序】本地老照片上色 AI 桌面工具

今日推荐：PyQt5 + ONNX 本地老照片上色（Image Colorization）AI 桌面工具

该工具基于深度学习图像着色模型（如 DDColor / UNet 类 Lab 通道预测模型），将黑白或灰度老照片自动还原为彩色图像。推理阶段完全使用ONNX Runtime，无需打包庞大的 PyTorch 依赖，适合离线分发。

理论说明

核心原理：主流图像上色模型通常在Lab 颜色空间工作。L 通道代表亮度（灰度信息），a、b 通道代表颜色信息。模型以 L 通道（1 通道灰度图）为输入，预测对应的 a、b 通道，再将三者拼接回 Lab 并转换为 RGB。

模型选择（可扩展）：DDColor（推荐）：基于双解码器结构，色彩还原更自然，支持转 ONNX。UNet / ResNet 类轻量模型：结构简单，推理快，适合低配 CPU。

ONNX 优势：将 PyTorch 动态图转为静态 ONNX 后，仅需 onnxruntime 即可推理，避免 PyInstaller 打包时陷入 PyTorch CUDA/C++ 扩展的依赖地狱。

环境或软件依赖

运行环境：Python 3.8+

核心依赖库：PyQt5（界面）onnxruntime（CPU 版推理，不要用 gpu 版以减少依赖）opencv-python（图像处理）numpy

模型文件：colorization.onnx（需自行将 PyTorch 预训练模型转 ONNX 放入 models/ 目录）

功能特性

•✅ 拖拽加载：直接拖拽图片到窗口即可加载

•✅ 模型切换：下拉菜单切换DDColor/UNet模型

•✅ 参数调节：饱和度和色调偏移实时调整

•✅ 进度显示：上色过程有进度条反馈

•✅ 高清导出：导出原始分辨率的PNG图片

•✅ 深色主题：护眼暗色界面

•✅ 异步处理：上色不阻塞界面操作

•✅ 中文路径：完美支持中文文件名和路径