| 1783 | |
| 1784 | @torch.jit.unused |
| 1785 | def _set_aux_loss(self, |
| 1786 | outputs_class, |
| 1787 | outputs_body_coord, |
| 1788 | outputs_lhand_coord, |
| 1789 | outputs_rhand_coord, |
| 1790 | outputs_face_coord, |
| 1791 | outputs_body_keypoints, |
| 1792 | outputs_lhand_keypoints, |
| 1793 | outputs_rhand_keypoints, |
| 1794 | outputs_face_keypoints, |
| 1795 | outputs_smpl_pose, |
| 1796 | outputs_smpl_rhand_pose, |
| 1797 | outputs_smpl_lhand_pose, |
| 1798 | outputs_smpl_jaw_pose, |
| 1799 | outputs_smpl_expr, |
| 1800 | outputs_smpl_beta, |
| 1801 | outputs_smpl_cam, |
| 1802 | # outputs_smpl_cam_f, |
| 1803 | outputs_smpl_kp3d): |
| 1804 | |
| 1805 | return [{ |
| 1806 | 'pred_logits': a, |
| 1807 | 'pred_boxes': b, |
| 1808 | 'pred_lhand_boxes': c, |
| 1809 | 'pred_rhand_boxes': d, |
| 1810 | 'pred_face_boxes': e, |
| 1811 | 'pred_keypoints': f, |
| 1812 | 'pred_lhand_keypoints': g, |
| 1813 | 'pred_rhand_keypoints': h, |
| 1814 | 'pred_face_keypoints': i, |
| 1815 | 'pred_smpl_pose': j, |
| 1816 | 'pred_smpl_rhand_pose': k, |
| 1817 | 'pred_smpl_lhand_pose': l, |
| 1818 | 'pred_smpl_jaw_pose': m, |
| 1819 | 'pred_smpl_expr': n, |
| 1820 | 'pred_smpl_beta': o, |
| 1821 | 'pred_smpl_cam': p, |
| 1822 | 'pred_smpl_kp3d': q |
| 1823 | } for a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q in zip( |
| 1824 | outputs_class[:-1], |
| 1825 | outputs_body_coord[:-1], |
| 1826 | outputs_lhand_coord[:-1], |
| 1827 | outputs_rhand_coord[:-1], |
| 1828 | outputs_face_coord[:-1], |
| 1829 | outputs_body_keypoints[:-1], |
| 1830 | outputs_lhand_keypoints[:-1], |
| 1831 | outputs_rhand_keypoints[:-1], |
| 1832 | outputs_face_keypoints[:-1], |
| 1833 | outputs_smpl_pose[:-1], |
| 1834 | outputs_smpl_rhand_pose[:-1], |
| 1835 | outputs_smpl_lhand_pose[:-1], |
| 1836 | outputs_smpl_jaw_pose[:-1], |
| 1837 | outputs_smpl_expr[:-1], |
| 1838 | outputs_smpl_beta[:-1], |
| 1839 | outputs_smpl_cam[:-1], |
| 1840 | outputs_smpl_kp3d[:-1])] |
| 1841 | |
| 1842 | def prepare_targets(self, data_batch): |