alexnasa commited on
Commit
be0edb3
·
verified ·
1 Parent(s): e0832eb

Update src/pixel3dmm/tracking/tracker.py

Browse files
Files changed (1) hide show
  1. src/pixel3dmm/tracking/tracker.py +5 -5
src/pixel3dmm/tracking/tracker.py CHANGED
@@ -343,7 +343,7 @@ class Tracker(object):
343
  v = vertices[b_i].cpu().numpy()
344
 
345
  if self.config.save_meshes:
346
- trimesh.Trimesh(faces=f, vertices=v, process=False).export(f'{self.mesh_folder}/{frame_id:05d}.ply')
347
  torch.save(frame, f'{self.checkpoint_folder}/{frame_id:05d}.frame')
348
 
349
  selction_indx = np.array([36, 39, 42, 45, 33, 48, 54])
@@ -356,12 +356,12 @@ class Tracker(object):
356
 
357
  if frame_id == self.config.start_frame and self.config.save_meshes:
358
  faces = self.diff_renderer.faces[0].cpu().numpy()
359
- trimesh.Trimesh(faces=faces, vertices=vertices_can[0].detach().cpu().numpy(), process=False).export(f'{self.mesh_folder}/canonical.ply')
360
  if self.config.save_landmarks:
361
  lmks = lmks.detach().squeeze().cpu().numpy()
362
- np.save(f'{self.mesh_folder}/ibug68_{frame_id}.ply', lmks)
363
  selction_indx = np.array([36, 39, 42, 45, 33, 48, 54])
364
- np.save(f'{self.mesh_folder}/now_{frame_id}.ply', lmks[selction_indx])
365
 
366
 
367
 
@@ -1289,7 +1289,7 @@ class Tracker(object):
1289
  # --- export the posed mesh, using the correct face indices field ---
1290
  os.makedirs(self.mesh_folder, exist_ok=True)
1291
  frame_id = str(0).zfill(5)
1292
- ply_path = os.path.join(self.mesh_folder, f"{frame_id}.ply")
1293
 
1294
  # pull out the face index tensor
1295
  faces_np = self.faces.verts_idx.cpu().numpy()
 
343
  v = vertices[b_i].cpu().numpy()
344
 
345
  if self.config.save_meshes:
346
+ trimesh.Trimesh(faces=f, vertices=v, process=False).export(f'{self.mesh_folder}/{frame_id:05d}.glb')
347
  torch.save(frame, f'{self.checkpoint_folder}/{frame_id:05d}.frame')
348
 
349
  selction_indx = np.array([36, 39, 42, 45, 33, 48, 54])
 
356
 
357
  if frame_id == self.config.start_frame and self.config.save_meshes:
358
  faces = self.diff_renderer.faces[0].cpu().numpy()
359
+ trimesh.Trimesh(faces=faces, vertices=vertices_can[0].detach().cpu().numpy(), process=False).export(f'{self.mesh_folder}/canonical.glb')
360
  if self.config.save_landmarks:
361
  lmks = lmks.detach().squeeze().cpu().numpy()
362
+ np.save(f'{self.mesh_folder}/ibug68_{frame_id}.glb', lmks)
363
  selction_indx = np.array([36, 39, 42, 45, 33, 48, 54])
364
+ np.save(f'{self.mesh_folder}/now_{frame_id}.glb', lmks[selction_indx])
365
 
366
 
367
 
 
1289
  # --- export the posed mesh, using the correct face indices field ---
1290
  os.makedirs(self.mesh_folder, exist_ok=True)
1291
  frame_id = str(0).zfill(5)
1292
+ ply_path = os.path.join(self.mesh_folder, f"{frame_id}.glb")
1293
 
1294
  # pull out the face index tensor
1295
  faces_np = self.faces.verts_idx.cpu().numpy()