Function convert_and_render

motion_gating/mbench_render.py:197–238

(
    motion_path: Path,
    motion_id: str,
    output_dir: Path,
    smplx_model_dir: Path,
    *,
    device: str,
    fps: int,
    width: int,
    height: int,
    noise_std: float = 0.0,
)

def convert_and_render(
    motion_path: Path,
    motion_id: str,
    output_dir: Path,
    smplx_model_dir: Path,
    *,
    device: str,
    fps: int,
    width: int,
    height: int,
    noise_std: float = 0.0,
) -> Tuple[Path, Path]:
    pt_path = output_dir / f"{motion_id}.pt"
    mp4_path = output_dir / f"{motion_id}.mp4"
    if pt_path.exists() and mp4_path.exists():
        return pt_path, mp4_path

    motion_tensor = load_motion_tensor(motion_path)

    smplx_model = SMPLX(
        model_path=str(smplx_model_dir),
        gender="neutral",
        use_pca=False,
        batch_size=motion_tensor.shape[0],
    ).to(device)
    smplx_model.eval()
    payload = run_smplx(motion_tensor, smplx_model, torch.device(device), noise_std=noise_std)
    output_dir.mkdir(parents=True, exist_ok=True)
    torch.save(payload, pt_path)

    render_video_from_vertices(
        payload["vertices"],
        payload["joints"],
        getattr(smplx_model, "faces").detach().cpu().numpy()
        if torch.is_tensor(getattr(smplx_model, "faces"))
        else np.asarray(getattr(smplx_model, "faces")),
        mp4_path,
        width=width,
        height=height,
        fps=fps,
    )
    return pt_path, mp4_path