save progress

2023-02-08 16:24:05 +00:00
parent 66e4212eab
commit ccec55d91f
6 changed files with 86602 additions and 45 deletions
@@ -223,9 +223,12 @@ def _todict(matobj):


 # can i cache map?
-def read_posed_garment_frame(folder, id, garment, frame, map, mask=None, shape=None, shapedirs=None, pose=None, unshape=False, deorient=False):
+def read_posed_garment_frame(folder, id, garment, frame, map=None, mask=None, shape=None, shapedirs=None, pose=None, unshape=False, deorient=False):
    pc16_path = os.path.join(folder, id, garment + '.pc16')
    verts = readPC2Frame(pc16_path, frame, True)
+
+    if map is None:
+        map = sio.loadmat(os.path.join(folder, id, garment + '_map.mat'))['map'].reshape((-1))
    verts = verts[map, :]

    if unshape: