diff --git a/convert_to_uv_local_single.py b/convert_to_uv_local_single.py index ea286fd..c536422 100644 --- a/convert_to_uv_local_single.py +++ b/convert_to_uv_local_single.py @@ -384,7 +384,7 @@ def test(verts, faces, values): if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/local-data/valid_skirt') + parser.add_argument("-f", "--folder", help = "end of range", default='/home/workspace/data/dataset/local-cache/train_skirt') parser.add_argument("-s", "--start", help = "start of range", default=0) parser.add_argument("-e", "--end", help = "end of range", default=30000) parser.add_argument("-n", "--name", help = "end of range", default='lhunor') @@ -442,17 +442,19 @@ if __name__ == '__main__': folder = args.folder frames_file = 'frames' + skirt_modifier + '_10p.txt' - neighbourhood = 10 + neighbourhood = 100 - # frames = [] - # with open(os.path.join(folder, frames_file), 'r') as f: - # lines = f.readlines() - # lines = [line.strip() for line in lines] - # frames = lines[int(args.start):int(args.end)] - # ids = [frame.split('_')[0] for frame in frames] + frames = [] + with open(os.path.join(folder, frames_file), 'r') as f: + lines = f.readlines() + lines = [line.strip() for line in lines] + frames = lines#[int(args.start):int(args.end)] + ids = [frame.split('_')[0] for frame in frames] - ids = os.listdir(os.path.join(local_folder, 'pose_shape')) - ids = [id.split('_')[0] for id in ids] + print(f'params: -f {args.folder} -s {args.start} -e {args.end} -n {args.name} -skirt {args.skirt} -neigh {neighbourhood}') + + # ids = os.listdir(os.path.join(local_folder, 'pose_shape')) + # ids = [id.split('_')[0] for id in ids] ids = sorted(list(set(ids)))[int(args.start):int(args.end)] for id in tqdm.tqdm(ids): diff --git a/run_parallel.py b/run_parallel.py index 428b431..9c623a2 100644 --- a/run_parallel.py +++ b/run_parallel.py @@ -4,7 +4,9 @@ import os if __name__ == '__main__': - folder = '/home/workspace/data/dataset/test_skirt_ready_2' + # folder = '/home/workspace/data/dataset/local-cache/train_skirt' + # folder = '/home/workspace/data/dataset/test_skirt_ready_2' + folder = '/home/workspace/local-data/valid_skirt' # folder = '/home/workspace/local-data/cape_processed/train/male' # folder = '/home/workspace/local-data/valid_skirt' # folder = '/home/workspace/local-data/valid' @@ -13,15 +15,15 @@ if __name__ == '__main__': skirt = 0 count = 0 - step = 25 + step = 10 # !!!!!!! change skirt parameter in script - while count < 200: + while count < 100: # os.system('nohup python convert_to_uv_cape.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_2_cape_train_male_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_2_cape_train_male_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python convert_to_uv.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_train_unposed_onehot_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_train_unposed_onehot_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python convert_to_uv_local.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_train_local_unposed_onehot_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_train_local_unposed_onehot_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') - os.system('nohup python convert_to_uv_local_single.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_curv_10_test_skirt_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_curv_10_test_skirt_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') + os.system('nohup python convert_to_uv_local_single.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_curv_100_valid_skirt_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_curv_100_valid_skirt_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python convert_to_uv_local_single.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_train_skirt_local_unposed_onehot_3_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_train_skirt_local_unposed_onehot_3_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python convert_to_uv_local_single.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'to_uv_valid_skirt_local_unposed_onehot_3_' + str(count) + '_' + str(count+step) + ' > nohup/to_uv_valid_skirt_local_unposed_onehot_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python downscale.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'downscale_valid_posed_no_orient_' + str(count) + '_' + str(count+step) + ' > nohup/downscale_valid_posed_no_orient_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &')