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))