在jupyterLab中数据集文件夹通常不能直接打包压缩,造成下载不便,因此需要一些工具函数进行辅助。
import os,zipfile
def pack2zip(path,zip_name):
#创建压缩文件夹
f = zipfile.ZipFile(zip_name,'w',zipfile.ZIP_DEFLATED)
#用os.walk来找到目录下每一个文件,再用os.path.join拼凑每个文件的完整路径,用write写入zip
for dir_path,dir_name,file_names in os.walk(path):
for file_name in file_names:
print(file_name)
f.write(os.path.join(dir_path,file_name))
f.close()
print('Done!')
#需要压缩的文件夹路径
path = 'data/images'
#压缩文件的名称
zip_name = 'a.zip'
pack2zip(path,zip_name)