-- rendering depth
(self, transformed_vertices)
| 468 | return shape_images |
| 469 | |
| 470 | def render_depth(self, transformed_vertices): |
| 471 | ''' |
| 472 | -- rendering depth |
| 473 | ''' |
| 474 | transformed_vertices = transformed_vertices.clone() |
| 475 | batch_size = transformed_vertices.shape[0] |
| 476 | |
| 477 | transformed_vertices[:, :, 2] = transformed_vertices[:, |
| 478 | :, 2] - transformed_vertices[:, :, 2].min() |
| 479 | z = -transformed_vertices[:, :, 2:].repeat(1, 1, 3) |
| 480 | z = z-z.min() |
| 481 | z = z/z.max() |
| 482 | # Attributes |
| 483 | attributes = util.face_vertices( |
| 484 | z, self.faces.expand(batch_size, -1, -1)) |
| 485 | # rasterize |
| 486 | rendering = self.rasterizer( |
| 487 | transformed_vertices, self.faces.expand(batch_size, -1, -1), attributes) |
| 488 | |
| 489 | #### |
| 490 | alpha_images = rendering[:, -1, :, :][:, None, :, :].detach() |
| 491 | depth_images = rendering[:, :1, :, :] |
| 492 | return depth_images |
| 493 | |
| 494 | def render_colors(self, transformed_vertices, colors, h=None, w=None): |
| 495 | ''' |
nothing calls this directly
no outgoing calls
no test coverage detected