Pytorch图像去噪实战（二十八）：TensorBoard可视化图像去噪训练过程，实时观察Loss、PSNR和去噪效果

Pytorch图像去噪实战（二十八）：TensorBoard可视化图像去噪训练过程，实时观察Loss、PSNR和去噪效果

一、问题场景：训练跑了半天，却不知道模型到底学得怎么样

训练图像去噪模型时，只打印 loss 很不够。

我以前经常只看：

Epoch 10, Loss=0.023 Epoch 20, Loss=0.018

看起来在下降，但模型到底有没有学好，其实不知道。

可能出现：

• loss下降但图像变糊
• PSNR提升但视觉效果不好
• 某个epoch开始过拟合
• 输出亮度慢慢偏移
• 噪声没去干净但指标看不出来

所以训练图像模型，一定要可视化。

本文使用 TensorBoard 记录：

• loss曲线
• PSNR指标
• noisy / pred / clean 对比图
• 学习率变化

二、安装依赖

pipinstalltensorboard

启动查看：<