Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-25 04:30:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove tinycudann for nerf

Browse Source
This commit is contained in:
xingjun.wxj
2025-03-12 10:43:33 +08:00
parent 3dff58b4d1
commit 3bfa5f4ce5
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/models/cv/nerf_recon_4k/network/utils.py
View File

@@ -5,7 +5,6 @@ from collections import defaultdict
import mcubes
import numpy as np
import tinycudann as tcnn
import torch
import torch.nn as nn
import torch.nn.functional as F
@@ -166,6 +165,8 @@ def normalize(dat, inp_scale, tgt_scale):
def cleanup():
import tinycudann as tcnn
gc.collect()
torch.cuda.empty_cache()
tcnn.free_temporary_memory()

3
modelscope/models/cv/nerf_recon_acc/network/nerf.py
View File

@@ -1,7 +1,6 @@
# The implementation is modified from nerfacc, made publicly available under the MIT License
# at https://github.com/KAIR-BAIR/nerfacc/blob/master/examples/radiance_fields/ngp.py
import numpy as np
import tinycudann as tcnn
import torch
import torch.nn as nn
from nerfacc import ContractionType, OccupancyGrid, ray_marching, rendering
@@ -113,6 +112,8 @@ trunc_exp = _TruncExp.apply
class VolumeDensity(nn.Module):
def __init__(self, config):
import tinycudann as tcnn
super().__init__()
self.config = config
self.radius = self.config.radius

3
modelscope/models/cv/nerf_recon_acc/network/utils.py
View File

@@ -5,7 +5,6 @@ from collections import defaultdict
import mcubes
import numpy as np
import tinycudann as tcnn
import torch
import torch.nn as nn
import torch.nn.functional as F
@@ -166,6 +165,8 @@ def normalize(dat, inp_scale, tgt_scale):
def cleanup():
import tinycudann as tcnn
gc.collect()
torch.cuda.empty_cache()
tcnn.free_temporary_memory()