Function plot_predictions

VAE_AutoEncoder/utils.py:76–89

:param y_true: 真实的图像 :param y_pred: 网络输出的图像 :return:

(y_true,y_pred,step,epoch)

    print("\n================================>Saving images...")

def plot_predictions(y_true,y_pred,step,epoch):
    """
    :param y_true: 真实的图像
    :param y_pred: 网络输出的图像
    :return:
    """
    f,ax = plt.subplots(nrows=2,ncols=10,sharex=True,sharey=True)
    for i in range(10):
        ax[0][i].imshow(np.reshape(y_true[i],(28,28)),aspect='auto')
        ax[1][i].imshow(np.reshape(y_pred[i].detach().cpu().numpy(),(28,28)),aspect='auto')
    #tight_layout会自动调整子图参数，使之填充整个图像区域。
    # 这是个实验特性，可能在一些情况下不工作。它仅仅检查坐标轴标签、刻度标签以及标题的部分。
    plt.tight_layout()
    plt.savefig(os.path.join(config.SAVE_IMAGES,str(epoch)+"_"+str(step)+'.png'))

def samples(args):
    """