update ae curv

2022-10-07 21:56:01 +00:00
parent feff6e315d
commit ece45943f8
3 changed files with 66 additions and 41 deletions
@@ -71,12 +71,12 @@ class UV2Mesh():
        

 class UV2MeshLabel():
-    def __init__(self, res=256):
+    def __init__(self, res=256, is_skirt=False):
        super(UV2MeshLabel, self).__init__()
        self.res = res
-        uv_map = np.load('custom_utils/stretched_uv_map_3.npy')
+        uv_map = np.load('custom_utils/stretched_uv_map_' + ('skirt' if is_skirt else '3') + '.npy')
        self.uv_map_discreet = np.floor(uv_map * self.res).astype(np.int32)
-        m2uv = np.load('custom_utils/m2uv.npy', allow_pickle=True).item()
+        m2uv = np.load('custom_utils/m2uv' + ('_skirt' if is_skirt else '') + '.npy', allow_pickle=True).item()
        m2uv = m2uv_array(m2uv)
        self.uv_pixels = self.uv_map_discreet[m2uv]