From 1b363a820e3a5595b14865ceca8e6c8f441ff370 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Thu, 8 Jun 2023 11:27:01 +0000 Subject: [PATCH] speed up rendering --- data/pose_dataset.py | 5 ++++- data/pose_renderer.py | 4 +++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/data/pose_dataset.py b/data/pose_dataset.py index 99d3951..c67fb0f 100644 --- a/data/pose_dataset.py +++ b/data/pose_dataset.py @@ -76,6 +76,9 @@ class Dataset(Dataset): body = self.smpl(pose.unsqueeze(0), shape.unsqueeze(0), remove=False) body_out = self.smpl(pose_out.unsqueeze(0), shape.unsqueeze(0), remove=False) + # print(f'smpl took {time() - start} seconds') + # start = time() + render = self._render_body(body[0])[0,:,:,:3] render_out = self._render_body(body_out[0])[0,:,:,:3] @@ -92,7 +95,7 @@ class Dataset(Dataset): render = render * mask render_out = render_out * mask_out - # print(f'one frame took {time() - start} seconds') + # print(f'render took {time() - start} seconds') data = { 'pose': render, diff --git a/data/pose_renderer.py b/data/pose_renderer.py index 4af8b40..fce3fda 100644 --- a/data/pose_renderer.py +++ b/data/pose_renderer.py @@ -47,12 +47,14 @@ class PoseRenderer(Module): image_size=(512, 512), blur_radius=0.0, faces_per_pixel=1, - bin_size=0 + bin_size=None, + max_faces_per_bin=None, ) renderer = MeshRenderer( rasterizer=MeshRasterizer(cameras=cameras, raster_settings=raster_settings), shader=HardPhongShader(device=verts.device, cameras=cameras, lights=AmbientLights(device=verts.device)) + # shader=SoftPhongShader(device=verts.device, cameras=cameras, lights=AmbientLights(device=verts.device)) ) output = renderer(Meshes(verts[None,:,:], faces[None,:,:], textures=self.textures))