demo.py 文件源码

def plot(result, i, directory = 'save'):
    pylab.figure(0)
    pylab.gray()
    plt.subplot(3,5,1)
    plt.axis('off')
    plt.imshow(result[0][:,:,(2,1,0)])
    plt.subplot(3,5,2)
    plt.axis('off')
    plt.imshow(result[1][:,:,(2,1,0)])
    plt.subplot(3,5,3)
    plt.axis('off')
    plt.imshow(result[2][:,:,(2,1,0)])
    plt.subplot(3,5,4)
    plt.axis('off')
    plt.imshow(result[3][:,:,(2,1,0)])
    plt.subplot(3,5,5)
    plt.axis('off')
    plt.imshow(result[4][:,:,(2,1,0)])
    plt.subplot(3,5,6)
    plt.axis('off')
    plt.imshow(result[5][:,:,(2,1,0)])
    plt.subplot(3,5,7)
    plt.axis('off')
    plt.imshow(result[6][:,:,(2,1,0)])
    plt.subplot(3,5,8)
    plt.axis('off')
    plt.imshow(result[7][:,:,(2,1,0)])
    plt.subplot(3,5,9)
    plt.axis('off')
    plt.imshow(result[8][:,:,(2,1,0)])
    plt.subplot(3,5,10)
    plt.axis('off')
    plt.imshow(result[9][:,:,(2,1,0)])
    plt.subplot(3,5,11)
    plt.axis('off')
    plt.imshow(result[10][:,:,(2,1,0)])
    plt.subplot(3,5,12)
    plt.axis('off')
    plt.imshow(result[11][:,:,(2,1,0)])
    plt.subplot(3,5,13)
    plt.axis('off')
    plt.imshow(result[12][:,:,(2,1,0)])
    plt.subplot(3,5,14)
    plt.axis('off')
    plt.imshow(result[13][:,:,(2,1,0)])
    plt.subplot(3,5,15)
    plt.axis('off')
    plt.imshow(result[14][:,:,(2,1,0)])
    plt.savefig(directory+'/'+str(i)+'.jpg')