invetigate registration errors

2023-03-30 14:13:44 +00:00
parent 898ad43316
commit 5ff4716033
3 changed files with 91 additions and 9 deletions
@@ -260,9 +260,12 @@ def read_posed_garment(folder, id, garment, map=None, mask=None, shape=None, sha
    return verts


-def read_garment_mask(folder, id, garment):
+def read_garment_mask(folder, id, garment, return_raw=False):
    raw_mask = sio.loadmat(os.path.join(folder, id, garment + '_mask.mat'))['mask'].reshape((-1))
-    mask = raw_mask > 0
+    if not return_raw:
+        mask = raw_mask > 0
+    else:
+        mask = raw_mask
    return mask