【 LPIPS + 颜色保真 + 像素级相似度 + 生成逼真度的超分 / 图像增强】

直接可用、带完整训练 / 测试 demo、同时用到 LPIPS + 颜色保真 + 像素级相似度 + 生成逼真度的超分 / 图像增强 Git 仓库

一、最强官方一站式(必试)

  1. BasicSR(XPixelGroup,顶会常用)
    https://github.com/XPixelGroup/BasicSR
    覆盖:超分 / 去噪 / 去模糊 / 色彩增强
    损失:LPIPS、SSIM、PSNR、MSE、Color Loss(Lab)、Perceptual Loss(VGG)、GAN Loss
    demo:scripts/train_sr.py、scripts/test_sr.py,配置文件直接开启所有 loss
    特点:模块化,可直接组合你要的 4 类损失
  2. Real-ESRGAN(xinntao,工业级)
    https://github.com/xinntao/Real-ESRGAN
    覆盖:真实世界超分、老照片修复、视频增强
    损失:LPIPS、VGG 感知损失、色彩一致性损失、像素 MSE、GAN 对抗损失
    demo:realesrgan/train.py,默认组合 LPIPS + 感知 + 像素损失
    特点:效果强,适合真实场景,含预训练模型
    二、科研向(论文复现,loss 齐全)
  3. ESRGAN(经典 baseline)
    https://github.com/xinntao/ESRGAN
    损失:LPIPS、VGG 感知损失、像素 MSE、GAN 损失
    demo:train.py,可直接添加颜色损失(Lab 空间)
    特点:SRGAN 升级版,感知质量高
  4. WAEP-SRGAN(LPIPS 增强版)
    https://github.com/shashankag14/Super-Resolution-GAN
    损失:LPIPS(核心)、MSE、SSIM、GAN、感知损失
    demo:完整训练脚本,默认加权组合所有 loss
    特点:解决 ESRGAN 偏色问题,颜色保真度好
  5. BSRGAN(盲超分 + 颜色增强)
    https://github.com/cszn/BSRGAN
    损失:LPIPS、SSIM、颜色损失(YCbCr/Lab)、感知损失
    demo:train_bsrgan.py,支持真实退化 + 色彩校正
    特点:盲超分强,颜色还原准
    三、轻量 / 扩散模型(新范式)
  6. FluxSR(华为 + 上交,2025)
    https://github.com/JianzeLi-114/FluxSR
    损失:LPIPS、TV Loss、感知损失、颜色一致性损失
    demo:训练脚本含完整 loss 配置,支持单步推理
    特点:速度快,细节与色彩平衡好
  7. PiSA-SR(双 LoRA,像素 / 语义解耦)
    https://github.com/xxxnell/PiSA-SR
    损失:LPIPS、PSNR、SSIM、CLIP 感知损失、颜色损失
    demo:可调节像素级(保真)与语义级(逼真)权重
    特点:可控制风格,适合个性化增强
    四、直接可用的 loss 组合模板(附代码)
    所有仓库都可直接加入这个4 合 1 loss:
    python
    运行

1. 像素级相似度(MSE/SSIM)

loss_pixel = F.mse_loss(pred, gt)

loss_ssim = 1 - kornia.losses.ssim_loss(pred, gt, 5)

2. 颜色保真度(Lab空间)

pred_lab = kornia.color.rgb_to_lab(pred)

gt_lab = kornia.color.rgb_to_lab(gt)

loss_color = F.l1_loss(pred_lab:, 1:, :, :, gt_lab:, 1:, :, :)

3. LPIPS感知相似度

loss_lpips = lpips_loss(pred, gt).mean()

4. 生成逼真度(VGG感知+GAN)

loss_perceptual = F.mse_loss(vgg(pred), vgg(gt))

loss_gan = gan_loss(discr(pred), torch.ones_like(discr(pred)))

总损失

total_loss = loss_pixel + 0.5loss_ssim + loss_color + loss_lpips + 0.1 loss_perceptual + 0.001*loss_gan

五、推荐优先级

BasicSR → 最全面,直接支持所有 loss,demo 成熟

Real-ESRGAN → 工业级效果,真实场景首选

BSRGAN → 颜色保真度强,盲超分友好

相关推荐
沈浩(种子思维作者)2 分钟前
心理的研究,我从何来?
数据库·人工智能·量子计算
AI原来如此3 分钟前
零基础教程:50页PDF一键浓缩成1页摘要
人工智能·ai·pdf·大模型
糖果店的幽灵7 分钟前
【langgraph 从入门到精通graphApi 篇】节点与边 —— 图的骨架
java·数据库·人工智能·redis·langgraph
理人综艺好会14 分钟前
ReAct Agent:AI智能体最基础的核心逻辑
人工智能
RUZHUA17 分钟前
2026企业媒体发稿平台选型指南:从资源效率到AI时代传播能力的全面评估
人工智能
minhuan18 分钟前
LangChain+LangGraph架构拆解:基于复杂AI任务编排与状态管理的大模型应用开发21.4
人工智能·langchain·大模型应用·langgraph·复杂ai任务编排
甲维斯27 分钟前
Kimi k3是真碉,但太不经用,也有拉跨点!
人工智能
KG_LLM图谱增强大模型29 分钟前
论文浅尝 | 基于先验思考的知识图谱上大语言模型的可信推理(NeurIPS 2025)
人工智能·语言模型·知识图谱
AI的探索之旅36 分钟前
我把立创EDA接进了本地大模型:数据不出门的AI设计检查实测
人工智能·stm32·硬件工程
hhzz36 分钟前
CNN猫狗图像分类实战:基于Keras+TensorFlow的卷积神经网络全流程解析
图像处理·python·深度学习·计算机视觉·cnn