plotting_utils.py 文件源码

def visualize_sequence(sequence, shape=(30, 40), savefilename=None, title='sequence'):
    cols = int(math.ceil(len(sequence) / 2.0))
    vis = tile_raster_images(sequence, shape, (2, cols), tile_spacing=(1, 1))
    plt.title(title)
    plt.axis('off')
    plt.show()
    if savefilename:
        o = Image.fromarray(vis)
        o.save('{}.png'.format(savefilename))