1.如果图像中出现大块的斑点,请检查上采样的过程,是否是使用了nn.convTranspose,连续的使用,而导致网络无法采集到全局信息而出现局部斑点。
改进方法:利用F.interpolate+conv2d配合取代原来的nn.convTranspose
2.如果图像的SSIM和PSNR参数一直处于很低的状态(SSIM远远小于90,PSNR远远小于40),则检查是否对图像数据进行归一化。
改进办法:对整个数据集进行最值寻找,然后进行归一化,具体的方法参考链接:
遍历整个文件夹中excel,寻找最值(归一化)-CSDN博客
PS:一定对寻找出的最值进行记录,因为后续需要反归一化
3.炼丹的过程中建议tensorboard和visdom两个图像可视化工具联合使用,以此明确是网络哪层的效果存在问题。
visdom使用时所遇的问题及解决方法_用visdom效果变差-CSDN博客
以上是目前对模型训练的心得,仅供参考