| 57 | import nvdiffrast.torch as dr |
| 58 | |
| 59 | def _warmup(glctx, device=None): |
| 60 | device = 'cuda' if device is None else device |
| 61 | #windows workaround for https://github.com/NVlabs/nvdiffrast/issues/59 |
| 62 | def tensor(*args, **kwargs): |
| 63 | return torch.tensor(*args, device=device, **kwargs) |
| 64 | pos = tensor([[[-0.8, -0.8, 0, 1], [0.8, -0.8, 0, 1], [-0.8, 0.8, 0, 1]]], dtype=torch.float32) |
| 65 | tri = tensor([[0, 1, 2]], dtype=torch.int32) |
| 66 | dr.rasterize(glctx, pos, tri, resolution=[256, 256]) |
| 67 | |
| 68 | class Pix2FacesRenderer: |
| 69 | def __init__(self, device="cuda"): |