use raw cloth3d input

data/preprocess.py

@@ -7,10 +7,11 @@ import scipy.io as sio
 from tqdm import tqdm
 
 sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
-from utils.utils import mask_faces
+from utils.data_utils import mask_faces, Frame
 from superSMPL.IO import readPC2, writePC2
 
-def generate_faces(folder):
+
+def generate_faces(folder, frames):
     frames = os.listdir(os.path.join(folder, 'mesh_mask'))
     F = np.load('custom_data/smpl_faces.npy')
     os.makedirs(os.path.join(folder, 'faces'), exist_ok=True)
@@ -21,12 +22,6 @@ def generate_faces(folder):
         np.save(os.path.join(folder, 'faces', frame), faces)
 
 
-class Frame:
-    def __init__(self, id, frame_nr):
-        self.id = id
-        self.frame_nr = frame_nr
-
-
 def generate_frames_file(folder, file_name):
     ids = os.listdir(folder)#[:1]
     frames = []
@@ -40,11 +35,11 @@ def generate_frames_file(folder, file_name):
         garment = garments[0]
         verts = readPC2(os.path.join(garment_dir, garment + '_registered.pc16'), True)['V']
         for i in range(verts.shape[0]):
-            frames.append(Frame(id, i))
+            frames.append(Frame(id, i, garment))
 
     with open(os.path.join(folder, file_name), 'w') as f:
         for i, frame in enumerate(frames):
-            f.write(str(frame.id) + ' ' + str(frame.frame_nr) + '\n')
+            f.write(str(frame.id) + ' ' + str(frame.frame_nr) + ' ' + str(frame.garment) + '\n')
 
 
 def genereated_registered_and_mask(folder, is_skirt = False):