From 28cdd79a6b042134860a2c6786d0fbb9570e752b Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 28 Jan 2023 20:25:31 +0000 Subject: [PATCH] generate test curv --- aaa_helpers.txt | 5 +++-- curvature_labels.py | 20 ++++++++++---------- 2 files changed, 13 insertions(+), 12 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 18b5361..296d4d4 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -1,4 +1,5 @@ -nohup python curvature_labels.py > nohup/curv_labels_train_skirt_15k_25k.out 2>&1 & +nohup python curvature_labels.py > nohup/curv_labels_test_skirt_0k_30k.out 2>&1 & +nohup python curvature_labels.py > nohup/curv_labels_valid_skirt_label_0k_30k.out 2>&1 & one final convolutional layer for smoothing, on the final coordinates @@ -6,7 +7,7 @@ skirts!!!! pyramid next week basic final, basic final skirts, - basic pyramid +basic pyramid removing curvatures test in ablation diff --git a/curvature_labels.py b/curvature_labels.py index 58b046a..6fcca26 100644 --- a/curvature_labels.py +++ b/curvature_labels.py @@ -251,9 +251,9 @@ def read_raw_curvatures(folder, frames, neighbourhood): if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument("-f", "--folder", help = "process the given folder", default=r'/home/workspace/data/dataset/local-cache/train_skirt') - parser.add_argument("-s", "--start", help = "start of range", default=15000) - parser.add_argument("-e", "--end", help = "end of range", default=25000) + parser.add_argument("-f", "--folder", help = "process the given folder", default=r'/home/workspace/data/dataset/test_skirt_ready_2') + 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") args = parser.parse_args() @@ -276,14 +276,14 @@ if __name__ == '__main__': print('SKIRT' if is_skirt else 'NOT SKIRT') print(f'params: -f {args.folder} -s {args.start} -e {args.end} -n {args.name} -skirt {is_skirt} -neigh {neighbourhood}') - # frames = sorted(os.listdir(os.path.join(folder, 'mesh_posed')))[int(args.start):int(args.end)] - # frames = [frame.split('.')[0] for frame in frames] + frames = sorted(os.listdir(os.path.join(folder, 'mesh_posed')))[int(args.start):int(args.end)] + frames = [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)] + # 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)] generate_curvatures(folder, neighbourhood, frames) # curvatures = read_raw_curvatures(folder, frames, neighbourhood)