可视化: 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]