Skip to content

Commit

Permalink
bugfix: remove coordinate normalization for MLP model
Browse files Browse the repository at this point in the history
  • Loading branch information
xuchen-ethz committed Jan 12, 2023
1 parent cecf85d commit f2b0c18
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions lib/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,8 +97,6 @@ def forward(self, pts_d, smpl_tfs, smpl_params, eval_mode=True, calc_time=False)
time['n_corres_bbox'] = mask[:,:num_point//2].float().sum().item()/(num_batch*num_point/2)
time['n_corres_surf'] = mask[:,num_point//2:].float().sum().item()/(num_batch*num_point/2)

pts_c = ((pts_c + self.deformer.offset) * self.deformer.scale+1)/2

occ_pd = self.network(pts_c, cond, mask=mask.reshape(num_batch*num_point * num_init)).reshape(num_batch, num_point, num_init)

if calc_time:
Expand Down

0 comments on commit f2b0c18

Please sign in to comment.