generate test curv

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

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)