如何查看重建图像

[shui-tian-ju-shi]

作者您好！我想查看模型训练完之后自我重建和交叉重建的图像效果。
我在train.py中加入了两行代码以加载训练好的模型。

我执行train.py文件，将网络预测的predictions['recontruction_imgs']里的四个值转化为rgb图像，使用utils包进行转换
from torchvision.utils import save_image

得到的重建图像有些异常

我不知道是什么原因导致的，预训练模型加载失败或者其他的什么原因，衷心地希望能得到您的建议，或者您能否分享一下代码？