可視化: Python—MatPlotLib—CV常用對比圖繪製

樣例

代碼：

plt.figure(figsize=(10, 2))
plt.subplots_adjust(bottom=0, left=.01, right=.99, top=.90, hspace=.35)
for idx, inp in enumerate(show_dogs):
    inp_origin = inp[0].numpy().transpose((1, 2, 0))
    inp_noise = inp[1].numpy().transpose((1, 2, 0))
    plt.subplot(2,10,idx+1)
    plt.imshow(inp_origin)
    plt.xticks(())
    plt.yticks(())
    plt.title("dog{}".format(idx))
    plt.subplot(2,10,idx+1+10)
    plt.imshow(inp_noise)
    plt.xticks(())
    plt.yticks(())

plt.figure(figsize=(10, 2))
plt.subplots_adjust(bottom=0, left=.01, right=.99, top=.90, hspace=.35)
for idx, inp in enumerate(show_cats):
    inp_origin = inp[0].numpy().transpose((1, 2, 0))
    inp_noise = inp[1].numpy().transpose((1, 2, 0))
    plt.subplot(2,10,idx+1)
    plt.imshow(inp_origin)
    plt.xticks(())
    plt.yticks(())
    plt.title("cat{}".format(idx))
    plt.subplot(2,10,idx+1+10)
    plt.imshow(inp_noise)
    plt.xticks(())
    plt.yticks(())

解釋

需要更改上方代碼的show_cats和show_dogs兩個變量。它們是代表圖像的tensor列表。具體的，每一個元素unit = [tensor1, tensor2]