def _create_mesh_model(self):
hmr = HumanModelRecovery(smpl_pkl_path=self._opt.smpl_model).cuda()
saved_data = torch.load(self._opt.hmr_model)
hmr.load_state_dict(saved_data)
hmr.eval()
print('load hmr model from {}'.format(self._opt.hmr_model))
return hmr
def _create_hmr(self):
hmr = HumanModelRecovery(smpl_pkl_path=self._opt.smpl_model)
saved_data = torch.load(self._opt.hmr_model)
hmr.load_state_dict(saved_data)
hmr.eval()
print('hmr load model from {} successfully.'.format(
self._opt.smpl_model))
return hmr
def _create_hmr(self):
hmr = HumanModelRecovery(self._opt.smpl_model)
saved_data = torch.load(self._opt.hmr_model)
hmr.load_state_dict(saved_data)
hmr.eval()
return hmr
