Super Resolution模型替换指南：自定义训练权重部署教程
1. 为什么需要替换超分模型？
你可能已经用过这个镜像，上传一张模糊的老照片，几秒钟后就拿到了3倍放大的高清图——细节清晰、边缘锐利、噪点消失。但某天你发现：对某些特定类型图片（比如手写字体、水墨画、低光照人像），效果不如预期；或者你想试试自己微调过的EDSR权重，又或者想换成Real-ESRGAN、SwinIR这类新架构……这时候，默认的EDSR_x3.pb就不再够用了。
这不是功能缺陷，而是设计初衷：它提供的是开箱即用的稳定服务，不是实验沙盒。而本教程要解决的，正是那个“下一步”——如何安全、可靠、不破坏现有WebUI的前提下，替换成你自己的超分模型权重。
整个过程不需要重装环境、不修改Flask服务逻辑、不碰OpenCV底层编译，只做三件事：确认模型格式兼容性、替换文件、重启服务。全程5分钟内完成，且支持随时回滚。
2. 先搞懂：当前系统用的是什么模型？
2.1 模型本质：OpenCV DNN SuperRes 接口
很多人误以为这是个“PyTorch模型部署”，其实不然。本镜像走的是OpenCV官方推荐的轻量级推理路径：
使用cv2.dnn_superres.DnnSuperResImpl_create()创建超分引擎
加载.pb（Protocol Buffer）格式的TensorFlow冻结图
调用upsample()方法完成推理
这意味着：你不能直接扔进去一个.pth或.onnx文件——它必须是OpenCV DNN模块能原生加载的格式。
2.2 当前模型关键信息
| 项目 | 值 | 说明 |
|---|---|---|
| 模型名称 | EDSR_x3.pb | 官方OpenCV贡献库中提供的预训练EDSR x3模型 |
| 存储路径 | /root/models/EDSR_x3.pb | 系统盘固化路径，重启不丢失 |
| 输入尺寸 | 任意（自动padding） | OpenCV SuperRes自动处理非整除尺寸 |
| 输出缩放倍数 | 固定x3 | 模型结构决定，不可运行时更改 |
| 推理后端 | TensorFlow CPU | 无需GPU，纯CPU即可流畅运行 |
** 重要提醒**：OpenCV DNN SuperRes目前仅支持4种官方模型（EDSR、ESPCN、FSRCNN、LapSRN），且每个模型都有严格对应的
.pb文件结构和输入输出张量名。自行转换模型时，必须确保输出节点名为output，输入节点名为input，否则会报错Can't create layer "input"。
3. 替换前必做：验证你的新模型是否兼容
别急着复制粘贴。90%的替换失败，都源于这一步跳过。
3.1 你需要准备的文件
- 一个
.pb格式的冻结图（TensorFlow 1.x导出） - 一份简短的Python验证脚本（我们帮你写好）
- 一张测试图（建议用镜像自带的
/root/test.jpg）
3.2 一行命令检查模型基础兼容性
打开终端，执行：
python3 -c "import cv2; sr = cv2.dnn_superres.DnnSuperResImpl_create(); sr.readModel('/root/models/your_model.pb'); print(' 模型加载成功');"如果看到模型加载成功，说明文件格式、节点命名、Opset基本过关。
如果报错，常见原因及修复方向如下：
| 报错信息 | 可能原因 | 解决思路 |
|---|---|---|
Can't create layer "input" | 输入节点名不是input | 用Netron打开.pb，重命名输入节点为input |
Can't create layer "output" | 输出节点名不是output | 同上，将最终输出节点重命名为output |
Unsupported op: ResizeBilinear | 含动态resize操作 | 改用静态上采样（如Conv2DTranspose）或转ONNX再转PB |
Failed to parse NetParameter file | PB文件损坏或版本不匹配 | 用TensorFlow 1.15导出，避免TF2.x SavedModel直转 |
3.3 运行端到端推理验证（推荐）
把下面这段代码保存为test_model.py，放在/root/目录下：
import cv2 import numpy as np # 1. 加载你的模型 sr = cv2.dnn_superres.DnnSuperResImpl_create() sr.readModel("/root/models/your_model.pb") sr.setModel("edsr", 3) # 注意：这里填模型类型名，不是文件名！可选 eds, espcn, fsrcnn, lapsrn # 2. 读取测试图（灰度或彩色均可） img = cv2.imread("/root/test.jpg") if img is None: print(" 测试图读取失败，请确认路径") exit(1) # 3. 执行超分 print("⏳ 正在推理...") result = sr.upsample(img) # 4. 保存结果并打印尺寸对比 cv2.imwrite("/root/test_result.jpg", result) h, w = img.shape[:2] hr, wr = result.shape[:2] print(f" 原图尺寸: {w}×{h} → 超分后: {wr}×{hr} (理论应为 {w*3}×{h*3})") print(" 结果已保存至 /root/test_result.jpg")运行它：
python3 /root/test_model.py- 如果输出尺寸接近
原宽×3 × 原高×3，且图片无严重伪影、色彩异常、全黑/全白，说明模型可用。 - 如果结果明显失真（如大面积色块、文字崩坏、边缘锯齿爆炸），请勿用于WebUI，需重新训练或调整后处理。
4. 正式替换：三步完成无缝切换
现在，你已确认新模型可用。接下来是生产环境安全替换。
4.1 备份原模型（强制步骤）
永远先备份。执行：
cp /root/models/EDSR_x3.pb /root/models/EDSR_x3.pb.bak_$(date +%Y%m%d_%H%M%S)你会在/root/models/下看到类似EDSR_x3.pb.bak_20240520_143022的备份文件。万一出问题，删掉新文件，重命名备份即可秒级回滚。
4.2 替换模型文件
将你验证通过的.pb文件（比如叫my_custom_edsr_x3.pb）上传到服务器，然后执行：
# 1. 复制到模型目录（注意目标文件名必须是 EDSR_x3.pb） cp /root/my_custom_edsr_x3.pb /root/models/EDSR_x3.pb # 2. 确保权限正确（OpenCV需要读取权限） chmod 644 /root/models/EDSR_x3.pb # 3. 查看文件大小是否合理（正常EDSR_x3.pb在30–45MB之间） ls -lh /root/models/EDSR_x3.pb** 小技巧**：如果你希望保留多个模型并快速切换，可以建立符号链接：
ln -sf /root/models/my_custom_edsr_x3.pb /root/models/EDSR_x3.pb这样只需改链接目标，无需反复复制大文件。
4.3 重启Web服务（关键！）
模型文件换了，但Flask服务还在用内存里的旧实例。必须重启：
# 停止当前服务（找到进程PID） ps aux | grep "flask" | grep -v "grep" # 假设PID是 1234，则执行： kill -9 1234 # 启动新服务（镜像默认启动命令） cd /root && python3 app.py** 镜像内服务启动脚本位置**：
/root/app.py
默认端口：5000（平台HTTP按钮即指向此端口）
日志查看：终端输出实时日志，关注是否出现Loading model from /root/models/EDSR_x3.pb字样。
等待几秒，点击平台HTTP按钮，上传同一张测试图——这次你看到的，就是你自己的模型效果了。
5. 进阶技巧：支持多模型热切换（可选）
如果你常需对比不同模型（比如EDSR vs Real-ESRGAN vs SwinIR），可以改造WebUI，实现下拉菜单选择：
5.1 修改模型加载逻辑（app.py）
找到app.py中模型初始化部分（通常在if __name__ == "__main__":上方），将硬编码改为字典管理：
# 替换原来的单模型加载 # sr = cv2.dnn_superres.DnnSuperResImpl_create() # sr.readModel("/root/models/EDSR_x3.pb") # sr.setModel("edsr", 3) # 改为多模型字典 MODEL_CONFIGS = { "edsr_x3": {"path": "/root/models/EDSR_x3.pb", "name": "edsr", "scale": 3}, "espcn_x3": {"path": "/root/models/ESPCN_x3.pb", "name": "espcn", "scale": 3}, "fsrcnn_x2": {"path": "/root/models/FSRCNN_x2.pb", "name": "fsrcnn", "scale": 2}, } # 初始化一个全局变量（实际项目建议用Flask g 或 cache） current_model_key = "edsr_x3" sr = cv2.dnn_superres.DnnSuperResImpl_create() sr.readModel(MODEL_CONFIGS[current_model_key]["path"]) sr.setModel(MODEL_CONFIGS[current_model_key]["name"], MODEL_CONFIGS[current_model_key]["scale"])5.2 添加前端切换控件（templates/index.html）
在上传按钮上方添加：
<div style="margin: 16px 0;"> <label for="model-select">选择超分模型：</label> <select id="model-select" onchange="changeModel(this.value)"> <option value="edsr_x3">EDSR x3（默认）</option> <option value="espcn_x3">ESPCN x3</option> <option value="fsrcnn_x2">FSRCNN x2</option> </select> </div> <script> function changeModel(key) { fetch(`/set_model?model=${key}`) .then(r => r.json()) .then(data => alert(" 模型已切换：" + data.model)); } </script>5.3 添加后端API路由（app.py）
在app.py中添加：
@app.route('/set_model') def set_model(): model_key = request.args.get('model', 'edsr_x3') if model_key not in MODEL_CONFIGS: return jsonify({"error": "模型不存在"}), 400 global sr, current_model_key sr = cv2.dnn_superres.DnnSuperResImpl_create() config = MODEL_CONFIGS[model_key] sr.readModel(config["path"]) sr.setModel(config["name"], config["scale"]) current_model_key = model_key return jsonify({"model": model_key})重启服务后，WebUI顶部就会出现下拉菜单，点击即切换，无需重启进程。
6. 总结：你已掌握超分模型自主权
回顾一下，你刚刚完成了什么：
- 理清了OpenCV SuperRes的模型加载机制，避开常见格式陷阱
- 学会了用两行Python验证新模型是否真正可用，而非盲目替换
- 实践了生产环境安全替换流程：备份→覆盖→重启，零风险
- 掌握了进阶方案：多模型热切换，让WebUI真正变成你的实验平台
这不再是“用别人调好的轮子”，而是你亲手校准、更换、驾驭的工具。下次看到一篇新的超分论文，你可以第一时间把它跑起来，而不是等别人打包好镜像。
超分辨率的本质，从来不只是“把图变大”，而是用AI重建被世界模糊掉的细节。而你现在，拥有了定义“细节”的权利。
获取更多AI镜像
想探索更多AI镜像和应用场景？访问 CSDN星图镜像广场，提供丰富的预置镜像，覆盖大模型推理、图像生成、视频生成、模型微调等多个领域，支持一键部署。
