generating more curvatures

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/test_skirt_ready_2')
+    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=5000)
-    parser.add_argument("-e", "--end", help = "end of range", default=10000)
+    parser.add_argument("-e", "--end", help = "end of range", default=15000)
     parser.add_argument("-n", "--name", help = "end of range", default="lhunor")
     args = parser.parse_args()
 
@@ -266,7 +266,7 @@ if __name__ == '__main__':
     frames_file = 'frames' + skirt_modifier + '_10p.txt'
 
     nr_of_labels = 6
-    neighbourhood = 10
+    neighbourhood = 100
     temp_file = '/home/workspace/local-data/temp/' + ''.join(random.choice(string.ascii_lowercase) for i in range(10)) + '.ply'
     if not is_skirt:
         dist = np.load('custom_data/dijkstra.npy')
@@ -275,13 +275,14 @@ if __name__ == '__main__':
 
     print('SKIRT' if is_skirt else 'NOT SKIRT')
 
-    frames = sorted(os.listdir(os.path.join(folder, 'mesh_posed')))[int(args.start):int(args.end)]
+    # 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)