Skip to content

Commit

Permalink
smplx has removed the dtype parameters
Browse files Browse the repository at this point in the history
  • Loading branch information
jonepatr committed Jan 12, 2021
1 parent bb0eb8e commit 99df432
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions FLAME.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ def _find_dynamic_lmk_idx_and_bcoords(self, vertices, pose, dynamic_lmk_faces_id
aa_pose = torch.index_select(pose.view(batch_size, -1, 3), 1,
neck_kin_chain)
rot_mats = batch_rodrigues(
aa_pose.view(-1, 3), dtype=dtype).view(batch_size, -1, 3, 3)
aa_pose.view(-1, 3)).view(batch_size, -1, 3, 3)

rel_rot_mat = torch.eye(3, device=vertices.device,
dtype=dtype).unsqueeze_(dim=0).expand(batch_size, -1, -1)
Expand Down Expand Up @@ -221,7 +221,7 @@ def forward(self, shape_params=None, expression_params=None, pose_params=None, n
vertices, _ = lbs(betas, full_pose, template_vertices,
self.shapedirs, self.posedirs,
self.J_regressor, self.parents,
self.lbs_weights, dtype=self.dtype)
self.lbs_weights)

lmk_faces_idx = self.lmk_faces_idx.unsqueeze(dim=0).repeat(
self.batch_size, 1)
Expand Down

0 comments on commit 99df432

Please sign in to comment.