Upload 7 files

visualise/rendering.py

@@ -130,6 +130,7 @@ class RenderTool():
             v[:, :, 1] = -v[:, :, 1]
             v[:, :, 2] = -v[:, :, 2]
         viewport_height = 800
+        z_offset = 1.0
         num_video = len(v_list)
         assert num_video in [1, 2, 3, 9, 12, 16, 18]
         if num_video == 1:
@@ -150,6 +151,7 @@ class RenderTool():
         if whole_body:
             width, height = 800, 1440
             viewport_height = 1440
+            z_offset = 1.8
 
         sr = 22000
         sr, audio = cur_wav_file
@@ -185,7 +187,7 @@ class RenderTool():
                     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((v_list[i][i_frame], self.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, z_offset=z_offset)
                 # sen = get_sen(i, num_video, i_frame, pos)
                 # if transcript is not None:
                 #     sen = str(int(transcript[i_frame].item()))

voca/rendering.py

@@ -40,7 +40,7 @@ def get_unit_factor(unit):
 
 
 def render_mesh_helper(mesh, t_center, rot=np.zeros(3), tex_img=None, v_colors=None,
-                       errors=None, error_unit='m', min_dist_in_mm=0.0, max_dist_in_mm=3.0, z_offset=0, xmag=0.5,
+                       errors=None, error_unit='m', min_dist_in_mm=0.0, max_dist_in_mm=3.0, z_offset=1.0, xmag=0.5,
                        y=0.7, z=1, camera='o', r=None):
     camera_params = {'c': np.array([0, 0]),
                      'k': np.array([-0.19816071, 0.92822711, 0, 0, 0]),
@@ -99,8 +99,8 @@ def render_mesh_helper(mesh, t_center, rot=np.zeros(3), tex_img=None, v_colors=N
     scene = pyrender.Scene(ambient_light=[.2, .2, .2], bg_color=[255, 255, 255])
 
     if camera == 'o':
-        camera = pyrender.OrthographicCamera(xmag=xmag, ymag=xmag)
-        # camera = pyrender.OrthographicCamera(xmag=xmag, ymag=xmag*1.8)
+        ymag = xmag * z_offset
+        camera = pyrender.OrthographicCamera(xmag=xmag, ymag=ymag)
     elif camera == 'i':
         camera = pyrender.IntrinsicsCamera(fx=camera_params['f'][0],
                                            fy=camera_params['f'][1],