Spaces:
Running
on
Zero
Running
on
Zero
Update src/pixel3dmm/tracking/tracker.py
Browse files
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}.
|
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.
|
360 |
if self.config.save_landmarks:
|
361 |
lmks = lmks.detach().squeeze().cpu().numpy()
|
362 |
-
np.save(f'{self.mesh_folder}/ibug68_{frame_id}.
|
363 |
selction_indx = np.array([36, 39, 42, 45, 33, 48, 54])
|
364 |
-
np.save(f'{self.mesh_folder}/now_{frame_id}.
|
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}.
|
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()
|