(self,
model_head,
body_model,
convention='smplx',
img_res=256,
scale_factor=2.0,
crop_size=256,
num_betas=10,
num_expression_coeffs=10,
condition_face_neck_pose=False,
condition_face_jaw_pose=True,
condition_face_shape=False,
condition_face_expression=True)
| 531 | position. |
| 532 | """ |
| 533 | def __init__(self, |
| 534 | model_head, |
| 535 | body_model, |
| 536 | convention='smplx', |
| 537 | img_res=256, |
| 538 | scale_factor=2.0, |
| 539 | crop_size=256, |
| 540 | num_betas=10, |
| 541 | num_expression_coeffs=10, |
| 542 | condition_face_neck_pose=False, |
| 543 | condition_face_jaw_pose=True, |
| 544 | condition_face_shape=False, |
| 545 | condition_face_expression=True): |
| 546 | self.model_head = model_head |
| 547 | self.body_model = body_model |
| 548 | self.img_res = img_res |
| 549 | self.convention = convention |
| 550 | self.num_betas = num_betas |
| 551 | self.num_expression_coeffs = num_expression_coeffs |
| 552 | |
| 553 | self.face_idx = get_keypoint_idxs_by_part('head', self.convention) |
| 554 | neck_idx = get_keypoint_idx('neck', self.convention) |
| 555 | self.neck_kin_chain = find_joint_kin_chain(neck_idx, |
| 556 | self.body_model.parents) |
| 557 | |
| 558 | self.condition_face_neck_pose = condition_face_neck_pose |
| 559 | self.condition_face_jaw_pose = condition_face_jaw_pose |
| 560 | self.condition_face_shape = condition_face_shape |
| 561 | self.condition_face_expression = condition_face_expression |
| 562 | |
| 563 | self.scale_factor = scale_factor |
| 564 | self.face_cropper = CropSampler(crop_size) |
| 565 | |
| 566 | def build_face_mean(self, global_orient, body_pose, betas, raw_jaw_pose, |
| 567 | expression, batch_size): |
nothing calls this directly
no test coverage detected