Upload 10 files

app.py

@@ -10,10 +10,10 @@ sys.path.append(os.getcwd())
 # os.system(r"apt-get update")
 # os.system(r"apt-get install sudo")
 #
-os.system(r"apt-get install libboost-dev")
-# os.system(r"sudo apt-get install gcc")
-# os.system(r"sudo apt-get install g++")
-os.system(r"make -C ./mesh-master all")
+# os.system(r"apt-get install libboost-dev")
+# # os.system(r"sudo apt-get install gcc")
+# # os.system(r"sudo apt-get install g++")
+# os.system(r"make -C ./mesh-master all")
 # os.system(r"cd ..")
 # os.system("pip install --no-deps --verbose --no-cache-dir /home/user/app/mesh-fix-MSVC_compilation")
 

visualise/rendering.py

@@ -11,7 +11,7 @@ import numpy as np
 from scipy.io import wavfile
 import pyrender
 
-from psbody.mesh import Mesh
+# from psbody.mesh import Mesh
 
 import librosa
 
@@ -95,11 +95,11 @@ def add_image_text(img, text, color=(0,0,255), w=800, h=800):
 
 class RenderTool():
     def __init__(self, out_path):
-        self.template_mesh = Mesh()
+        # self.template_mesh = Mesh()
         path = os.path.join(os.getcwd(), 'visualise/smplx/SMPLX_NEUTRAL.npz')
         model_data = np.load(path, allow_pickle=True)
         data_struct = Struct(**model_data)
-        self.template_mesh.f = data_struct.f
+        self.f = data_struct.f
         self.out_path = out_path
         if not os.path.exists(self.out_path):
             os.makedirs(self.out_path)
@@ -182,10 +182,10 @@ class RenderTool():
             cur_img = []
             for i in range(len(v_list)):
                 if face:
-                    img = render_mesh_helper(Mesh(v_list[i][i_frame], self.template_mesh.f), center,
+                    img = render_mesh_helper((v_list[i][i_frame], self.f), center,
                                                   r=r, xmag=0.15, y=1, z=1.0, camera='o')
                 else:
-                    img = render_mesh_helper(Mesh(v_list[i][i_frame], self.template_mesh.f), center, camera='o',r=r, y=0.7)
+                    img = render_mesh_helper((v_list[i][i_frame], self.f), center, camera='o',r=r, y=0.7)
                 # sen = get_sen(i, num_video, i_frame, pos)
                 # if transcript is not None:
                 #     sen = str(int(transcript[i_frame].item()))

voca/rendering.py

@@ -25,7 +25,7 @@ import tempfile
 import numpy as np
 import matplotlib as mpl
 import matplotlib.cm as cm
-from psbody.mesh import Mesh
+# from psbody.mesh import Mesh
 
 
 def get_unit_factor(unit):
@@ -48,8 +48,8 @@ def render_mesh_helper(mesh, t_center, rot=np.zeros(3), tex_img=None, v_colors=N
 
     frustum = {'near': 0.01, 'far': 3.0, 'height': 800, 'width': 800}
 
-    mesh_copy = Mesh(mesh.v, mesh.f)
-    mesh_copy.v[:] = cv2.Rodrigues(rot)[0].dot((mesh_copy.v - t_center).T).T + t_center
+    v, f = mesh
+    v = cv2.Rodrigues(rot)[0].dot((v - t_center).T).T + t_center
 
     texture_rendering = tex_img is not None and hasattr(mesh, 'vt') and hasattr(mesh, 'ft')
     if texture_rendering:
@@ -86,7 +86,7 @@ def render_mesh_helper(mesh, t_center, rot=np.zeros(3), tex_img=None, v_colors=N
     color = np.array([0.3, 0.5, 0.55])
 
     if not texture_rendering:
-        tri_mesh = trimesh.Trimesh(vertices=mesh_copy.v, faces=mesh_copy.f, vertex_colors=rgb_per_v)
+        tri_mesh = trimesh.Trimesh(vertices=v, faces=f, vertex_colors=rgb_per_v)
         render_mesh = pyrender.Mesh.from_trimesh(tri_mesh,
                                                  smooth=True,
                                                  material=pyrender.MetallicRoughnessMaterial(