generate overfit jumpsuit results

curvature_analysis.py

@@ -205,19 +205,25 @@ def visualize_curvature_labels(folder, frame, neighbourhoods):
 
 if __name__ == '__main__':
 
-    is_skirt = True
+    is_skirt = False
     skirt_modifier = '_skirt' if is_skirt else ''
     F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32)
     # k_neighbours(k=25)
 
-    folder = '/home/workspace/local-data/valid_skirt'
+    folder = '/home/workspace/local-data/train'
+    results_folder = ''
+
+    frames_file = 'frames_jumpsuit_30k.txt'
+    frames = []
+    with open(os.path.join(folder, frames_file), 'r') as f:
+        lines = f.readlines()
+        lines = [line.strip() for line in lines]
+    frames = lines
 
-    nr_of_labels = 6
-    neighbourhood = 50
     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')
     else:
         dist = np.load('custom_data/dijkstra_skirt.npy')
 
-    visualize_curvature_labels(folder, '02572_1', neighbourhoods=[10, 25, 50, 100])
+    visualize_curvature_labels(folder, '00005_1', neighbourhoods=[10, 25, 50, 100])