本项目旨在实现:高清原图 + 已翻译但画质差的图 => 高清已翻译图片。
原理:训练模型生成对话框的蒙版(MASK),调用PS自动批量执行。
| 原图 | 识别对话框 | 应用蒙版 |
|---|---|---|
| 图片匹配 | PSD生成 |
|---|---|
需要提前安装PS,本人使用的是PS2022,其他较新的版本理论上也可以(未测试)
请先学习PS的基础知识,现阶段对某些气泡框识别效果不佳,建议使用画笔工具手动微调蒙版区域以改善成图效果。
小白推荐的视频:图层与蒙版最通俗易懂的讲解
下载打包好的文件(仅可调用),解压直接运行start.bat
注意:请下载v1.8打包版,然后下载最新源码覆盖。
下载本项目,安装环境(可重新训练模型),然后运行:
pip install -r requirements.txt
python web.py如果生成的requirements有缺漏,那就自行安装吧。
注意:如果pillow-avif-plugin安装时编译失败,可用去找镜像源里找现成的whl包。
- 可观看此简易教程:Tutorial.mp4 (观看者限18岁以上)
- 运行后,使用Edge或Chrome浏览器访问:http://127.0.0.1:5000
- 输入文件夹路径
- 点击
开始匹配按钮,等待匹配结果 - 检查匹配结果,如果匹配不对,请在设置按钮中调整
匹配阈值,或手动点击匹配项的取消按钮,或从手动匹配栏拖动图片,或点击“信任”按钮强制将匹配率置1 - 关闭各种滤镜,点击“选一张试试”以生成一个预览的PSD,在此PSD中尝试调整各滤镜参数以使两图层网点或色彩更接近
- 设置生成PSD时的参数
- 确定好
匹配结果旁边的数字,这是发送任务到PS执行的图片组,规则是“匹配图不为空+匹配率大于阈值” - 点击
发送任务到PS,会在原图目录下生成一个auto_PSD文件夹,生成的PSD存放于此 - 定期清理
temp_mask文件夹,这是MASK的临时目录
- 训练模型以生成气泡框MASK图层
- 使用Resnet50提取图片特征,以匹配图片组
- 生成PSD时自动对齐图层(使用的是PS自带的功能/cv2特征点匹配,目前感觉cv2效果更好)
- 生成PSD时判断是否为黑白图,如果是,可选择性转为灰度图
- 在前端控制对图层调整或使用“色阶”、“表面模糊”、“USM锐化”、“指定的动作”。
注意:滤镜顺序是“表面模糊—色阶—USM锐化—指定的动作” - 可选择性使用“尝试自动对齐颜色”(依据图片明亮度调整),效果一般
如果想要自己训练模型,需准备好数据集,这里提供本人标记的数据集PSD,需使用其他脚本/导出图片.py这个脚本导出为png。
数据集下载地址:manga-remake-dataset(使用者限18岁以上,且仅供学习使用,下载后请于24小时内删除)
- 2026-01-11 数据集数量:1332(英文530张 + 日文802张)
- 代码苦手,感谢豆包和Gork~
- 没有设计美感,UI凑合着用吧