Simplest way to visualize a sequence of T pose.
(num_frames,
body_model_config=None,
body_model=None,
orbit_speed=1.0,
**kwargs)
| 1149 | |
| 1150 | |
| 1151 | def visualize_T_pose(num_frames, |
| 1152 | body_model_config=None, |
| 1153 | body_model=None, |
| 1154 | orbit_speed=1.0, |
| 1155 | **kwargs) -> None: |
| 1156 | """Simplest way to visualize a sequence of T pose.""" |
| 1157 | assert num_frames > 0, '`num_frames` is required.' |
| 1158 | assert body_model_config is not None or body_model is not None |
| 1159 | model_type = body_model_config[ |
| 1160 | 'type'] if body_model_config is not None else body_model.name( |
| 1161 | ).replace('-', '').lower() |
| 1162 | if model_type == 'smpl': |
| 1163 | poses = torch.zeros(num_frames, 72) |
| 1164 | else: |
| 1165 | poses = torch.zeros(num_frames, 165) |
| 1166 | |
| 1167 | func = partial(render_smpl, |
| 1168 | betas=None, |
| 1169 | transl=None, |
| 1170 | verts=None, |
| 1171 | convention='pytorch3d', |
| 1172 | projection='fovperspective', |
| 1173 | K=None, |
| 1174 | R=None, |
| 1175 | T=None, |
| 1176 | origin_frames=None) |
| 1177 | for k in func.keywords.keys(): |
| 1178 | if k in kwargs: |
| 1179 | kwargs.pop(k) |
| 1180 | return func(poses=poses, |
| 1181 | body_model_config=body_model_config, |
| 1182 | body_model=body_model, |
| 1183 | orbit_speed=orbit_speed, |
| 1184 | **kwargs) |
| 1185 | |
| 1186 | |
| 1187 | def visualize_smpl_pose(poses=None, verts=None, **kwargs) -> None: |