PLT多图对比展示

写了，顺手记录一下

# ref
# https://zhzhx.blog.csdn.net/article/details/115859955
#https://blog.csdn.net/a19990412/article/details/81407701
import cv2
import matplotlib.pyplot as plt
import matplotlib.image as Image

# 使用matplotlib展示多张图片

def matplotlib_multi_pic1(scal):
    ori = []
    bicubic = []
    rrdngans = []
    rdnsmall = []
    rdnnoi = []

    if scal == '2x':
        i2 = 4
        for i in range(9):
            ori.append(Image.imread('./data/downsample/32x32/32x32_1 ({}).jpg'.format(i+1)))
            bicubic.append(Image.imread('./data/upsample/bicubic/x2_16x16/bicubic_x2_1 ({}).jpg'.format(i + 1)))
            rdnsmall.append(Image.imread('./data/upsample/ISR/rdn_psnr_small/x2_16x16/psnr_small_1 ({}).jpeg'.format(i + 1)))
            rdnnoi.append(Image.imread('./data/upsample/ISR/rdn_psnr_noise-cancel/x2_16x16/psnr_noise-cancel_x2_1 ({}).jpeg'.format(i + 1)))

        list = [ori, bicubic, rdnsmall, rdnnoi]
    if scal == '4x':
        i2 = 3
        for i in range(9):
            ori.append(Image.imread('./data/downsample/64x64/1 ({}).jpg'.format(i+1)))
            bicubic.append(Image.imread('./data/upsample/bicubic/x4_16x16/bicubic_x4_1 ({}).jpg'.format(i + 1)))
            rrdngans.append(Image.imread('./data/upsample/ISR/rrdn_gans/x4_16x16/rrdn_gans_1 ({}).jpeg'.format(i + 1)))
        list = [ori,bicubic,rrdngans]

    for i in range(i2):
        # title = "Ori_11111"
        # plt.title(title, fontsize=6)
        for j in range(9):
            if i == 0:
                title="Ori_"+str(j+1)
            elif i == 1:
                title = "Bicubic_" + str(j + 1)
            elif i == 2:
                if scal == '2x':
                    title = "Rdn_small_" + str(j + 1)
                elif scal == '4x':
                    title = "Rrdn_gans_" + str(j + 1)
            elif i == 3:
                title = "Rdn_noise_" + str(j + 1)

            #行，列，索引
            plt.subplot(i2,9,i*9+j+1)
            plt.imshow(list[i][j])
            plt.title(title,fontsize=6)
            plt.xticks([])
            plt.yticks([])

    plt.savefig("{}.jpg".format(scal))
    plt.show()
def origin():
    ori = []
    for i in range(9):
        ori.append(Image.imread('./data/ori/1 ({}).jpg'.format(i + 1)))



        title = str(i + 1)
        plt.subplot(3, 3, i+1)
        plt.imshow(ori[i])
        plt.title(title, fontsize=6)
        plt.xticks([])
        plt.yticks([])
    plt.savefig("origin.jpg")
    plt.show()


    pass

# matplotlib_multi_pic1(scal='4x')
origin()