From 42599ee4650bd08d2dcfaddecf752d3ac86746f8 Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Mon, 15 Nov 2021 16:50:35 +0100 Subject: [PATCH] Add support for starting syn data generation from specific point --- create_syn_data.sh | 2 +- data/create_syn_data.py | 12 +++++++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/create_syn_data.sh b/create_syn_data.sh index 356514d..35573b2 100755 --- a/create_syn_data.sh +++ b/create_syn_data.sh @@ -4,4 +4,4 @@ cd data/lcn python setup.py build_ext --inplace cd ../ -python create_syn_data.py +python create_syn_data.py $2 diff --git a/data/create_syn_data.py b/data/create_syn_data.py index d637fd7..9ffac74 100644 --- a/data/create_syn_data.py +++ b/data/create_syn_data.py @@ -218,6 +218,16 @@ if __name__=='__main__': out_root = data_root / f'{data_type}' out_root.mkdir(parents=True, exist_ok=True) + start = 0 + if len(sys.argv) >= 2 and isinstance(sys.argv[2], int): + start = sys.argv[2] + else: + if sys.argv[2] == '--resume': + try: + start = max([int(dir) for dir in os.listdir(out_root) if str.isdigit(dir)]) or 0 + except: + pass + # load shapenet models obj_classes = ['chair'] objs = get_objs(shapenet_root, obj_classes) @@ -254,6 +264,6 @@ if __name__=='__main__': # start the job n_samples = 2**10 + 2**13 - for idx in range(n_samples): + for idx in range(start, n_samples): args = (out_root, idx, n_samples, imsize, patterns, K, baseline, blend_im, noise, track_length) create_data(*args)