python——打包img到 zip

方法1：

import matplotlib.pyplot as plt, zipfile 
from PIL import Image 
import os

z = zipfile.PyZipFile('images.zip', mode='w')

for k in range(100):
    img = Image.fromarray( np.uint8(imgdata) )
    f = str(k)+'.png'
    img.save(f,'PNG'); z.write(f); os.remove(f)
    
z.close()

方法2：

import shutil

from torchvision.utils import save_image

 

for i in range(100):
        save_image(images[i, :, :, :], os.path.join('../output_images', f'image_{i}.png'))

shutil.make_archive('images', 'zip', '../output_images')    
    

 

 

提取方法：

 with zipfile.ZipFile("../images.zip","r") as z:
        z.extractall("../tmp/")