nodes/MrForExample-ComfyUI-3D-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove a few not too useful nodes

Browse Source 
Remove NeuS, DMTet
Refactor GSConverterNeRFMarchingCubes
This commit is contained in:
Mr. For Example
2024-07-03 15:32:40 -06:00
parent 2d356dc0cc
commit 44545fcc15
11 changed files with 2247 additions and 938 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mesh_processer/mesh_utils.py
+6 -5
View File
@@ -494,14 +494,15 @@ def marching_cubes_density_to_mesh(get_density_func, grid_size=256, S=128, densi
return vertices, triangles
def color_func_to_albedo(mesh, get_rgb_func, texture_resolution=1024, padding=2, batch_size=640000, device="cuda", force_cuda_rast=False):
def color_func_to_albedo(mesh, get_rgb_func, texture_resolution=1024, padding=2, batch_size=640000, device="cuda", force_cuda_rast=False, glctx=None):
import nvdiffrast.torch as dr
from kiui.op import uv_padding
if force_cuda_rast:
glctx = dr.RasterizeCudaContext()
else:
glctx = dr.RasterizeGLContext()
if glctx is None:
if force_cuda_rast:
glctx = dr.RasterizeCudaContext()
else:
glctx = dr.RasterizeGLContext()
# render uv maps
h = w = texture_resolution