|
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) |
|
|
|
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) |
|
|
|
|