Spaces:
Sleeping
Sleeping
## Pytorch3D Renderer | |
> The code was modified from GVHMR: https://github.com/zju3dv/GVHMR/tree/main/hmr4d/utils/vis | |
### Dependency | |
```shell | |
pip install "git+https://github.com/facebookresearch/[email protected]" | |
``` | |
### Example | |
```python | |
from lib.utils.vis import Renderer | |
import imageio | |
fps = 30 | |
focal_length = data["cam_int"][0][0, 0] | |
width, height = img_hw | |
faces = smplh[data["gender"]].bm.faces | |
renderer = Renderer(width, height, focal_length, "cuda", faces) | |
writer = imageio.get_writer("tmp_debug.mp4", fps=fps, mode="I", format="FFMPEG", macro_block_size=1) | |
for i in tqdm(range(length)): | |
img = np.zeros((height, width, 3), dtype=np.uint8) | |
img = renderer.render_mesh(smplh_out.vertices[i].cuda(), img) | |
writer.append_data(img) | |
writer.close() | |
``` |