mnist_save_images.py 文件源码

def save_images(images, dir):
    save_path = base_dir + dir + '_28'
    os.makedirs(save_path)
    prefix = dir
    for i, img_data in enumerate(images):
        img_data2 = skimage.img_as_ubyte(img_data)
        img = Image.fromarray(img_data2)
        img.save('%s/%s_%d.tiff' % (save_path, prefix, i))