alexjc/neural-enhance

Neural Enhance镜像说明

镜像概述

Neural Enhance是一款基于深度学习的图像与视频超分辨率工具，通过预训练模型或自定义训练，实现图像的放大（2x/4x）、JPEG压缩瑕疵修复、去噪等功能，帮助提升低分辨率图像的细节与清晰度，适用于多种专业与日常图像处理场景。

核心功能

• 图像超分辨率：支持2x/4x放大，基于训练模型生成细节丰富的高清图像；
• 图像修复：修复JPEG压缩产生的 artifacts，还原图像原本质量；
• 自定义训练：可基于自有数据集训练专属模型，提升特定场景下的处理效果；
• 多设备支持：兼容CPU与GPU运行，GPU版本需配合nvidia-docker实现加速。

使用场景

• 老照片修复：提升老照片的分辨率与清晰度，还原历史细节；
• 低清图像增强：将低分辨率图像转为高清，适用于设计、摄影等领域；
• 视频帧处理：对视频帧进行超分，提升视频整体质量；
• 批量图像处理：支持多图像同时处理，提高工作效率。

配置说明

Docker部署示例

基本使用

bash
# 下载镜像并查看帮助
docker run --rm -v `pwd`:/ne/input -it alexjc/neural-enhance --help

设置别名（简化操作）

将以下命令添加到.bashrc或.zshrc中，方便快速调用：

bash
alias enhance='function ne() { docker run --rm -v "$(pwd)/`dirname ${@:$#}`":/ne/input -it alexjc/neural-enhance ${@:1:$#-1} "input/`basename ${@:$#}`"; }; ne'

处理单张图像

bash
# 修复JPEG artifacts（无放大）
enhance --zoom=1 --model=repair images/broken.jpg
# 2x放大高质量图像
enhance --zoom=2 images/high_quality.jpg

处理多张图像

bash
# 批量处理文件夹内所有JPG图像（2x放大）
enhance --zoom=2 "images/*.jpg"

GPU加速使用

需安装nvidia-docker，使用GPU版本镜像：

bash
# 设置GPU版本别名
alias enhance-gpu='function ne() { docker run --rm --gpus all -v "$(pwd)/`dirname ${@:$#}`":/ne/input -it alexjc/neural-enhance:gpu ${@:1:$#-1} "input/`basename ${@:$#}`"; }; ne'
# 使用GPU处理图像
enhance-gpu --zoom=2 images/sample.jpg

注意事项

• 自定义训练需准备足够的数据集，调整参数以获得最佳效果；
• GPU版本需确保CUDA环境配置正确，且nvidia-docker正常运行；
• 处理大尺寸图像时，需保证设备内存充足。