| ''' |
| Clean all output files produced by this project. |
| ''' |
|
|
| from utils.config import get_dataset, get_args |
| import os |
| from tqdm import tqdm |
|
|
| def delete_files(dataset): |
| dirs_to_be_deleted = [f'{dataset.root}/output'] |
| for dir_to_be_deleted in dirs_to_be_deleted: |
| if os.path.exists(dir_to_be_deleted): |
| os.system(f'rm -r {dir_to_be_deleted}') |
|
|
| def main(dataset): |
| delete_files(dataset) |
|
|
| def process_one_dataset(split_path, args): |
| with open(split_path, 'r') as f: |
| seq_name_list = f.readlines() |
| seq_name_list = [seq_name.strip() for seq_name in seq_name_list] |
| for seq_name in tqdm(seq_name_list): |
| args.seq_name = seq_name |
| dataset = get_dataset(args) |
| main(dataset) |
|
|
| if __name__ == '__main__': |
| args = get_args() |
| |
| |
| split_path = 'splits/scannet.txt' |
| args.dataset = 'scannet' |
| process_one_dataset(split_path, args) |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
