save

preprocessing.py

@@ -216,7 +216,7 @@ def generate_faces():
         np.save(os.path.join(output_folder, id + '.npy'), faces.astype(np.int32))
 
 
-def generate_frames_file(folder, ratio=0.1):
+def generate_frames_file(folder, ratio=0.1, garment='', file_name='frames.txt'):
     all_frames = os.listdir(os.path.join(folder, 'pose_shape'))
     ids = [frame.split('_')[0] for frame in all_frames]
     all_frames = [frame.split('.')[0] for frame in all_frames]
@@ -232,17 +232,15 @@ def generate_frames_file(folder, ratio=0.1):
     selected = []
 
     for id in tqdm.tqdm(sub_ids):
-        if garment_type_table[id] != 'Dress':
+        if garment_type_table[id] != garment:
             continue
-        if len(selected) > 30000:
-            break
 
         # frames = glob.glob(os.path.join(folder, 'pose_shape', id + '_*'))
         frames = [frame for frame in all_frames if frame.startswith(id + '_')]
         frames = [frame.split('/')[-1].split('.')[0] for frame in frames]
         selected.extend(frames)
 
-    with open(os.path.join(folder, 'frames_dress_30k.txt'), 'w') as f:
+    with open(os.path.join(folder, file_name), 'w') as f:
         for i, frame in enumerate(selected):
             f.write(frame + '\n')
 
@@ -293,7 +291,7 @@ def generate_frames_file_filtered_cape():
 
 if __name__ == '__main__':
 
-    src_folder = '/home/workspace/local-data/train_skirt' #cloth3d_skirt
+    src_folder = '/home/workspace/local-data/train' #cloth3d_skirt
     # src_folder = r'D:\Work\Dataset\validation_2'
     dest_folder = '/home/workspace/data/dataset/validation' 
 
@@ -306,5 +304,17 @@ if __name__ == '__main__':
     # verify_poses(src_folder)
     # create_garment_type_table()
     # generate_faces()
-    generate_frames_file(src_folder, ratio=1)
+    # generate_frames_file(src_folder, ratio=1)
     # generate_frames_file_filtered_cape()
+
+
+    # # folders = ['/home/workspace/local-data/train', '/home/workspace/local-data/test_ready', '/home/workspace/local-data/valid']
+    # folders = ['/home/workspace/local-data/valid_skirt']
+    # # garments = ['Jumpsuit', 'Trousers', 'Tshirt', 'Top']
+    # # filename = ['frames_jumpsuit.txt', 'frames_trousers.txt', 'frames_tshirt.txt', 'frames_top.txt']
+    # garments = ['Skirt', 'Dress', 'Tshirt', 'Top']
+    # filename = ['frames_skirt.txt', 'frames_dress.txt', 'frames_tshirt.txt', 'frames_top.txt']
+
+    # for folder in folders:
+    #     for garment, file in zip(garments, filename):
+    #         generate_frames_file(folder, garment=garment, file_name=file, ratio=1)