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启动查看: