(self, render_size=1024, config=None, device='cuda:0', part='body', background=None, rasterizer_type='standard')
| 39 | ''' |
| 40 | |
| 41 | def __init__(self, render_size=1024, config=None, device='cuda:0', part='body', background=None, rasterizer_type='standard'): |
| 42 | if config is None: |
| 43 | self.cfg = cfg |
| 44 | else: |
| 45 | self.cfg = config |
| 46 | self.device = device |
| 47 | self.render_size = render_size |
| 48 | # cache data for smplx texture |
| 49 | self.smplx_texture = imread(self.cfg.model.smplx_tex_path)/255. |
| 50 | self.cached_data = np.load( |
| 51 | self.cfg.model.flame2smplx_cached_path, allow_pickle=True, encoding='latin1').item() |
| 52 | # Set up the renderer for visualization |
| 53 | self.part = part |
| 54 | self.rasterizer_type = rasterizer_type |
| 55 | self._setup_renderer() |
| 56 | # background |
| 57 | if background is not None: |
| 58 | self.bg = background |
| 59 | else: |
| 60 | self.bg = torch.ones( |
| 61 | [1, 1, self.render_size, self.render_size], device=self.device, dtype=torch.float32) |
| 62 | |
| 63 | def _setup_renderer(self): |
| 64 | # setup raterizer |
nothing calls this directly
no test coverage detected