get ids

data/preprocessing.py

@@ -766,16 +766,50 @@ def double_check_irregular_poses(folder, folder_reg, out_folder):
         assert body_count == verts_count
 
 
+def get_ids():
+    ids = []
+    with open(os.path.join('/data-local/127/cloth3d_processed', 'frames_train.txt'), 'r') as f:
+        for line in f:
+            ids.append(line.split()[0])
+
+    with open(os.path.join('/data-local/127/cloth3d_processed', 'frames_train_skirt.txt'), 'r') as f:
+        for line in f:
+            ids.append(line.split()[0])
+
+    ids = sorted(list(set(ids)))
+
+    with open(os.path.join('/data-local/127/cloth3d_processed', 'frames_train_ids.txt'), 'w') as f:
+        for id in ids:
+            f.write(f'{id}\n')
+
+    ids = []
+    with open(os.path.join('/data-local/127_2/cloth3d_processed_test', 'frames_test.txt'), 'r') as f:
+        for line in f:
+            ids.append(line.split()[0])
+
+    with open(os.path.join('/data-local/127_2/cloth3d_processed_test', 'frames_test_skirt.txt'), 'r') as f:
+        for line in f:
+            ids.append(line.split()[0])
+
+    ids = sorted(list(set(ids)))
+
+    with open(os.path.join('/data-local/127_2/cloth3d_processed_test', 'frames_test_ids.txt'), 'w') as f:
+        for id in ids:
+            f.write(f'{id}\n')
+
+    pass
+
+
 if __name__ == '__main__':
 
 
-    folder = '/data-net/dataset/cloth3d/test'
-    folder_reg = '/data-local/127/cloth3d_registration/test'
-    out_folder = '/data-local/127_2/cloth3d_processed_test'
+    # folder = '/data-net/dataset/cloth3d/test'
+    # folder_reg = '/data-local/127/cloth3d_registration/test'
+    # out_folder = '/data-local/127_2/cloth3d_processed_test'
 
-    # folder = '/data-net/dataset/cloth3d/train'
-    # folder_reg = '/data-local/127/cloth3d_registration/train'
-    # out_folder = '/data-local/127/cloth3d_processed'
+    folder = '/data-net/dataset/cloth3d/train'
+    folder_reg = '/data-local/127/cloth3d_registration/train'
+    out_folder = '/data-local/127/cloth3d_processed'
 
     smpl = SMPL(path='smpl', device='cuda')
 
@@ -798,7 +832,8 @@ if __name__ == '__main__':
     # calculate_average(folder, out_folder)
 
     # create_frames_file(folder, out_folder)
-    create_frames_file_test(folder, out_folder)
+    get_ids()
+    # create_frames_file_test(folder, out_folder)
     # create_frames_file_partial(folder, out_folder, part=0.5)
     
     # folder = '/data-local/127_2/cape_processed/test/female'