| |
|
|
| from . import _init_paths |
| import numpy as np |
| import Sim3DR_Cython |
|
|
|
|
| def get_normal(vertices, triangles): |
| normal = np.zeros_like(vertices, dtype=np.float32) |
| Sim3DR_Cython.get_normal(normal, vertices, triangles, vertices.shape[0], triangles.shape[0]) |
| return normal |
|
|
|
|
| def rasterize(vertices, triangles, colors, bg=None, |
| height=None, width=None, channel=None, |
| reverse=False): |
| if bg is not None: |
| height, width, channel = bg.shape |
| else: |
| assert height is not None and width is not None and channel is not None |
| bg = np.zeros((height, width, channel), dtype=np.uint8) |
|
|
| buffer = np.zeros((height, width), dtype=np.float32) - 1e8 |
|
|
| if colors.dtype != np.float32: |
| colors = colors.astype(np.float32) |
| Sim3DR_Cython.rasterize(bg, vertices, triangles, colors, buffer, triangles.shape[0], height, width, channel, |
| reverse=reverse) |
| return bg |
|
|
