(sample, extrinsics, intrinsics, options={}, colors_overwrite=None, verbose=True, **kwargs)
| 70 | |
| 71 | |
| 72 | def render_frames(sample, extrinsics, intrinsics, options={}, colors_overwrite=None, verbose=True, **kwargs): |
| 73 | renderer = get_renderer(sample, **options) |
| 74 | rets = {} |
| 75 | for j, (extr, intr) in tqdm(enumerate(zip(extrinsics, intrinsics)), desc='Rendering', disable=not verbose): |
| 76 | if isinstance(sample, MeshExtractResult): |
| 77 | res = renderer.render(sample, extr, intr) |
| 78 | if 'normal' not in rets: rets['normal'] = [] |
| 79 | rets['normal'].append(np.clip(res['normal'].detach().cpu().numpy().transpose(1, 2, 0) * 255, 0, 255).astype(np.uint8)) |
| 80 | else: |
| 81 | res = renderer.render(sample, extr, intr, colors_overwrite=colors_overwrite) |
| 82 | if 'color' not in rets: rets['color'] = [] |
| 83 | if 'depth' not in rets: rets['depth'] = [] |
| 84 | rets['color'].append(np.clip(res['color'].detach().cpu().numpy().transpose(1, 2, 0) * 255, 0, 255).astype(np.uint8)) |
| 85 | if 'percent_depth' in res: |
| 86 | rets['depth'].append(res['percent_depth'].detach().cpu().numpy()) |
| 87 | elif 'depth' in res: |
| 88 | rets['depth'].append(res['depth'].detach().cpu().numpy()) |
| 89 | else: |
| 90 | rets['depth'].append(None) |
| 91 | return rets |
| 92 | |
| 93 | |
| 94 | def render_video(sample, resolution=512, bg_color=(0, 0, 0), num_frames=300, r=2, fov=40, **kwargs): |
no test coverage detected