(self, humans, pred_rend_array, verts_list=None, color_list=None)
| 297 | return normal_map[0, :, :, 0] |
| 298 | |
| 299 | def render_frame(self, humans, pred_rend_array, verts_list=None, color_list=None): |
| 300 | if not isinstance(pred_rend_array, np.ndarray): |
| 301 | pred_rend_array = np.asarray(pred_rend_array) |
| 302 | self.renderer.create_camera(self.default_R, self.default_T) |
| 303 | _img = pred_rend_array |
| 304 | if humans is not None: |
| 305 | for human in humans: |
| 306 | _img = self.renderer.render_mesh(human['v3d'].to(self.device), _img) |
| 307 | else: |
| 308 | for i, verts in enumerate(verts_list): |
| 309 | if color_list is None: |
| 310 | _img = self.renderer.render_mesh(verts.to(self.device), _img) |
| 311 | else: |
| 312 | _img = self.renderer.render_mesh(verts.to(self.device), _img, color_list[i]) |
| 313 | if self.keep_origin: |
| 314 | _img = np.concatenate([np.asarray(pred_rend_array), _img],1).astype(np.uint8) |
| 315 | return _img |
| 316 | |
| 317 | def render_video(self, results, pil_bis_frames, fps, out_path): |
| 318 | writer = imageio.get_writer( |
no test coverage detected