File size: 1,216 Bytes
f53b39e |
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 |
# Copyright (C) 2022-present Naver Corporation. All rights reserved.
# Licensed under CC BY-NC-SA 4.0 (non-commercial use only).
"""
Script generating commandlines to generate image pairs from metadata files.
"""
import os
import glob
from tqdm import tqdm
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--input_dir", required=True)
parser.add_argument("--output_dir", required=True)
parser.add_argument("--prefix", default="", help="Commanline prefix, useful e.g. to setup environment.")
args = parser.parse_args()
input_metadata_filenames = glob.iglob(f"{args.input_dir}/**/metadata.json", recursive=True)
for metadata_filename in tqdm(input_metadata_filenames):
output_dir = os.path.join(args.output_dir, os.path.relpath(os.path.dirname(metadata_filename), args.input_dir))
# Do not process the scene if the metadata file already exists
if os.path.exists(os.path.join(output_dir, "metadata.json")):
continue
commandline = f"{args.prefix}python datasets/habitat_sim/generate_from_metadata.py --metadata_filename={metadata_filename} --output_dir={output_dir}"
print(commandline)
|