save progress

data/curvatures.py

@@ -127,10 +127,10 @@ class Cache:
             if self.include_neighbours:
                 k_neighbors = np.argsort(dist[mask, :][:, mask], axis=-1)[:, :self.neighbourhood]
                 self.cache[id].k_neighbors = k_neighbors
-                G = tm.graph.vertex_adjacency_graph(m)
-                for i in range(v.shape[0]):
-                    for k in range(25):
-                        assert nx.has_path(G, i, k_neighbors[i, k]), f'No path between {i} and {k_neighbors[i, k]}'
+                # G = tm.graph.vertex_adjacency_graph(m)
+                # for i in range(v.shape[0]):
+                #     for k in range(25):
+                #         assert nx.has_path(G, i, k_neighbors[i, k]), f'No path between {i} and {k_neighbors[i, k]}'
 
             self.cache[id].mask = mask
             self.cache[id].faces = faces
@@ -331,11 +331,12 @@ def read_raw_curvatures_2(folder, frames, neighbourhood):
     
     return curvatures_all
 
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument("-f", "--folder", help = "process the given folder", default=r'/home/workspace/data/dataset/cloth3d/train')
     parser.add_argument("-s", "--start", help = "start of range", default=0)
-    parser.add_argument("-e", "--end", help = "end of range", default=1000)
+    parser.add_argument("-e", "--end", help = "end of range", default=600000)
     parser.add_argument("-n", "--name", help = "end of range", default="lhunor")
     args = parser.parse_args()