(smpl_joints_batch, title_batch=None, outname=None)
| 136 | |
| 137 | |
| 138 | def draw_to_batch(smpl_joints_batch, title_batch=None, outname=None): |
| 139 | |
| 140 | batch_size = len(smpl_joints_batch) |
| 141 | out = [] |
| 142 | for i in range(batch_size): |
| 143 | out.append( |
| 144 | plot_3d_motion([ |
| 145 | smpl_joints_batch[i], None, |
| 146 | title_batch[i] if title_batch is not None else None |
| 147 | ])) |
| 148 | if outname is not None: |
| 149 | imageio.mimsave(outname[i], np.array(out[-1]), duration=50) |
| 150 | out = torch.stack(out, axis=0) |
| 151 | return out |
nothing calls this directly
no test coverage detected