tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

TensoRF go brrrrrrr (#1609)

Browse Source 
* pdfsampler no longer includes uniform samples

* Lint

* Typo fix

---------

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
This commit is contained in:
David McAllister 2023-03-15 20:27:37 -07:00 committed by GitHub
parent af29ead3a2
commit 4c2e200098
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/configs/method_configs.py
View File

@ -310,6 +310,8 @@ method_configs["tensorf"] = TrainerConfig(
pipeline=VanillaPipelineConfig(
datamanager=VanillaDataManagerConfig(
dataparser=BlenderDataParserConfig(),
train_num_rays_per_batch=4096,
eval_num_rays_per_batch=4096,
),
model=TensoRFModelConfig(),
),

9
nerfstudio/models/tensorf.py
View File

@ -70,8 +70,10 @@ class TensoRFModelConfig(ModelConfig):
"""specifies a list of iteration step numbers to perform upsampling"""
loss_coefficients: Dict[str, float] = to_immutable_dict({"rgb_loss": 1.0})
"""Loss specific weights."""
num_samples: int = 256
num_samples: int = 50
"""Number of samples in field evaluation"""
num_uniform_samples: int = 200
"""Number of samples in density evaluation"""
num_den_components: int = 16
"""Number of components in density encoding"""
num_color_components: int = 48
@ -116,7 +118,6 @@ class TensoRFModel(Model):
def get_training_callbacks(
self, training_callback_attributes: TrainingCallbackAttributes
) -> List[TrainingCallback]:
# the callback that we want to run every X iterations after the training iteration
def reinitialize_optimizer(
self, training_callback_attributes: TrainingCallbackAttributes, step: int # pylint: disable=unused-argument
@ -219,8 +220,8 @@ class TensoRFModel(Model):
)
# samplers
self.sampler_uniform = UniformSampler(num_samples=self.config.num_samples, single_jitter=True)
self.sampler_pdf = PDFSampler(num_samples=self.config.num_samples // 2, single_jitter=True)
self.sampler_uniform = UniformSampler(num_samples=self.config.num_uniform_samples, single_jitter=True)
self.sampler_pdf = PDFSampler(num_samples=self.config.num_samples, single_jitter=True, include_original=False)
# renderers
self.renderer_rgb = RGBRenderer(background_color=colors.WHITE)