| # Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT | |
| # except for the third-party components listed below. | |
| # Hunyuan 3D does not impose any additional limitations beyond what is outlined | |
| # in the repsective licenses of these third-party components. | |
| # Users must comply with all terms and conditions of original licenses of these third-party | |
| # components and must ensure that the usage of the third party components adheres to | |
| # all relevant laws and regulations. | |
| # For avoidance of doubts, Hunyuan 3D means the large language models and | |
| # their software and algorithms, including trained model weights, parameters (including | |
| # optimizer states), machine-learning model code, inference-enabling code, training-enabling code, | |
| # fine-tuning enabling code and other elements of the foregoing made publicly available | |
| # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT. | |
| import custom_rasterizer_kernel | |
| import torch | |
| def rasterize(pos, tri, resolution, clamp_depth=torch.zeros(0), use_depth_prior=0): | |
| assert pos.device == tri.device | |
| findices, barycentric = custom_rasterizer_kernel.rasterize_image( | |
| pos[0], tri, clamp_depth, resolution[1], resolution[0], 1e-6, use_depth_prior | |
| ) | |
| return findices, barycentric | |
| def interpolate(col, findices, barycentric, tri): | |
| f = findices - 1 + (findices == 0) | |
| vcol = col[0, tri.long()[f.long()]] | |
| result = barycentric.view(*barycentric.shape, 1) * vcol | |
| result = torch.sum(result, axis=-2) | |
| return result.view(1, *result.shape) | |