| 142 | return pose_meta |
| 143 | |
| 144 | def load_from_meta(self, meta, norm_body=True, norm_hand=False): |
| 145 | self.image_id = meta.get("image_id", "00000.png") |
| 146 | self.height = meta["height"] |
| 147 | self.width = meta["width"] |
| 148 | kps_body_p = [] |
| 149 | kps_body = [] |
| 150 | for kp in meta["keypoints_body"]: |
| 151 | if kp is None: |
| 152 | kps_body.append([0, 0]) |
| 153 | kps_body_p.append(0) |
| 154 | else: |
| 155 | kps_body.append(kp) |
| 156 | kps_body_p.append(1) |
| 157 | |
| 158 | self.kps_body = np.array(kps_body) |
| 159 | self.kps_body[:, 0] *= self.width |
| 160 | self.kps_body[:, 1] *= self.height |
| 161 | self.kps_body_p = np.array(kps_body_p) |
| 162 | |
| 163 | self.kps_lhand = np.array(meta["keypoints_left_hand"])[:, :2] |
| 164 | self.kps_lhand_p = np.array(meta["keypoints_left_hand"])[:, 2] |
| 165 | self.kps_rhand = np.array(meta["keypoints_right_hand"])[:, :2] |
| 166 | self.kps_rhand_p = np.array(meta["keypoints_right_hand"])[:, 2] |
| 167 | |
| 168 | @staticmethod |
| 169 | def load_from_kp2ds(kp2ds: List[np.ndarray], width: int, height: int): |