luni/reconstruction
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

speed up rendering

Browse Source
This commit is contained in:
luni
2023-06-08 11:27:01 +00:00
parent 882ddf5578
commit 1b363a820e
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/pose_dataset.py
+4 -1
View File
@@ -76,6 +76,9 @@ class Dataset(Dataset):
body = self.smpl(pose.unsqueeze(0), shape.unsqueeze(0), remove=False) body = self.smpl(pose.unsqueeze(0), shape.unsqueeze(0), remove=False)
body_out = self.smpl(pose_out.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 = self._render_body(body[0])[0,:,:,:3]
render_out = self._render_body(body_out[0])[0,:,:,:3] render_out = self._render_body(body_out[0])[0,:,:,:3]
@@ -92,7 +95,7 @@ class Dataset(Dataset):
render = render * mask render = render * mask
render_out = render_out * mask_out render_out = render_out * mask_out
# print(f'one frame took {time() - start} seconds') # print(f'render took {time() - start} seconds')
data = { data = {
'pose': render, 'pose': render,
data/pose_renderer.py
+3 -1
View File
@@ -47,12 +47,14 @@ class PoseRenderer(Module):
image_size=(512, 512), image_size=(512, 512),
blur_radius=0.0, blur_radius=0.0,
faces_per_pixel=1, faces_per_pixel=1,
bin_size=0 bin_size=None,
max_faces_per_bin=None,
) )
renderer = MeshRenderer( renderer = MeshRenderer(
rasterizer=MeshRasterizer(cameras=cameras, raster_settings=raster_settings), rasterizer=MeshRasterizer(cameras=cameras, raster_settings=raster_settings),
shader=HardPhongShader(device=verts.device, cameras=cameras, lights=AmbientLights(device=verts.device)) 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)) output = renderer(Meshes(verts[None,:,:], faces[None,:,:], textures=self.textures))