fixed normalization, can overfit on single image

data/preprocess.py

@@ -9,6 +9,7 @@ from tqdm import tqdm
 sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
 from utils.data_utils import mask_faces, Frame
 from superSMPL.IO import readPC2, writePC2
+from cloth3d.util import loadInfo
 
 
 def generate_faces(folder, frames):
@@ -71,7 +72,7 @@ def genereated_registered_and_mask(folder, is_skirt = False):
 if __name__ == '__main__':
     # folder = '/home/workspace/local-data/train'
     # folder = '/home/workspace/local-data/valid'
-    folder = 'dataset/CLOTH3D/train'
+    folder = '/home/workspace/data/dataset/cloth3d/train'
 
     # generate_faces(folder)
     # genereated_registered_and_mask(folder)