File size: 1,443 Bytes
cc9780d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import os
import glob
import argparse
parser = argparse.ArgumentParser("unzip the prepared data")
parser.add_argument("--occ_root", type=str, default="../data/occ_data")
parser.add_argument("--other_root", type=str,default="../data/other_data")
parser.add_argument("--unzip_occ",default=False,action="store_true")
parser.add_argument("--unzip_other",default=False,action="store_true")
args=parser.parse_args()
if args.unzip_occ:
filelist=os.listdir(args.occ_root)
for filename in filelist:
filepath=os.path.join(args.occ_root,filename)
if ".rar" in filename:
unrar_command="unrar x %s %s"%(filepath,args.occ_root)
os.system(unrar_command)
elif ".zip" in filename:
unzip_command="7z x %s -o%s"%(filepath,args.occ_root)
os.system(unzip_command)
if args.unzip_other:
category_list=os.listdir(args.other_root)
for category in category_list:
category_folder=os.path.join(args.other_root,category)
#print(category_folder)
rar_filelist=glob.glob(category_folder+"/*.rar")
zip_filelist=glob.glob(category_folder+"/*.zip")
for rar_filepath in rar_filelist:
unrar_command="unrar x %s %s"%(rar_filepath,category_folder)
os.system(unrar_command)
for zip_filepath in zip_filelist:
unzip_command="7z x %s -o%s"%(zip_filepath,category_folder)
os.system(unzip_command)
|