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