def generate_objects(hmmr_out): folder_path = model_folder model = SMPLModel("assets/basicmodel_m_lbs_10_207_0_v1.0.0.pkl") counter = 0 for i in hmmr_out["verts"]: model.verts = i model.save_to_obj(folder_path + "/" + str(counter) + ".obj") counter += 1
def save_generated_smpl(self, epoch, examples=3, render=True): """ Save generated SMPL parameters """ if self.ex_data is None: self.ex_data = np.random.randn(examples, self._gen_dim) generated_params = self.generator.predict(self.ex_data) for i, params in enumerate(generated_params): np.save( "../gen_pose_params/gan_example.E{:03d}:{:02d}.npy".format( epoch, i), params) if render: smpl = SMPLModel('../model.pkl') beta = np.zeros(shape=(10, )) trans = np.zeros(shape=(3, )) for i, pose_params in enumerate(generated_params): smpl.set_params(beta=beta, pose=pose_params, trans=trans) smpl.save_to_obj('../meshes/smpl_np.E{:03d}:{:02d}.obj'.format( epoch, i))