Method
_set_aux_loss
(self,
outputs_class,
outputs_body_coord,
outputs_lhand_coord,
outputs_rhand_coord,
outputs_face_coord,
outputs_smpl_pose,
outputs_smpl_rhand_pose,
outputs_smpl_lhand_pose,
outputs_smpl_jaw_pose,
outputs_smpl_expr,
outputs_smpl_beta,
outputs_smpl_cam,
outputs_smpl_kp3d)
Source from the content-addressed store, hash-verified
| 3574 | |
| 3575 | @torch.jit.unused |
| 3576 | def _set_aux_loss(self, |
| 3577 | outputs_class, |
| 3578 | outputs_body_coord, |
| 3579 | outputs_lhand_coord, |
| 3580 | outputs_rhand_coord, |
| 3581 | outputs_face_coord, |
| 3582 | outputs_smpl_pose, |
| 3583 | outputs_smpl_rhand_pose, |
| 3584 | outputs_smpl_lhand_pose, |
| 3585 | outputs_smpl_jaw_pose, |
| 3586 | outputs_smpl_expr, |
| 3587 | outputs_smpl_beta, |
| 3588 | outputs_smpl_cam, |
| 3589 | outputs_smpl_kp3d): |
| 3590 | |
| 3591 | return [{ |
| 3592 | 'pred_logits': a, |
| 3593 | 'pred_boxes': b, |
| 3594 | 'pred_lhand_boxes': c, |
| 3595 | 'pred_rhand_boxes': d, |
| 3596 | 'pred_face_boxes': e, |
| 3597 | 'pred_smpl_pose': j, |
| 3598 | 'pred_smpl_rhand_pose': k, |
| 3599 | 'pred_smpl_lhand_pose': l, |
| 3600 | 'pred_smpl_jaw_pose': m, |
| 3601 | 'pred_smpl_expr': n, |
| 3602 | 'pred_smpl_beta': o, |
| 3603 | 'pred_smpl_cam': p, |
| 3604 | 'pred_smpl_kp3d': q |
| 3605 | } for a, b, c, d, e, j, k, l, m, n, o, p, q in zip( |
| 3606 | outputs_class[:-1], |
| 3607 | outputs_body_coord[:-1], |
| 3608 | outputs_lhand_coord[:-1], |
| 3609 | outputs_rhand_coord[:-1], |
| 3610 | outputs_face_coord[:-1], |
| 3611 | outputs_smpl_pose[:-1], |
| 3612 | outputs_smpl_rhand_pose[:-1], |
| 3613 | outputs_smpl_lhand_pose[:-1], |
| 3614 | outputs_smpl_jaw_pose[:-1], |
| 3615 | outputs_smpl_expr[:-1], |
| 3616 | outputs_smpl_beta[:-1], |
| 3617 | outputs_smpl_cam[:-1], |
| 3618 | outputs_smpl_kp3d[:-1])] |
| 3619 | |
| 3620 | def prepare_targets(self, data_batch): |
| 3621 | |
Tested by
no test coverage detected