From 4059cbaddbf6fa3e7eefd314a1d6b2e014a68496 Mon Sep 17 00:00:00 2001 From: terrance Date: Thu, 15 Dec 2022 08:34:24 +0000 Subject: [PATCH] remove unused sampler, normalize predicted normals from network --- nerfstudio/fields/dreamfusion_field.py | 3 ++- nerfstudio/models/dreamfusion.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/fields/dreamfusion_field.py b/nerfstudio/fields/dreamfusion_field.py index dee66a71d..ca5afc803 100644 --- a/nerfstudio/fields/dreamfusion_field.py +++ b/nerfstudio/fields/dreamfusion_field.py @@ -43,6 +43,7 @@ from nerfstudio.field_components.spatial_distortions import ( SpatialDistortion, ) from nerfstudio.fields.base_field import Field +from nerfstudio.utils import colormaps, colors, math try: import tinycudann as tcnn @@ -208,7 +209,7 @@ class DreamFusionField(Field): pred_normals_inp = torch.cat([positions_flat, density_embedding.view(-1, self.geo_feat_dim)], dim=-1) x = self.mlp_pred_normals(pred_normals_inp).view(*outputs_shape, -1).to(directions) - outputs[FieldHeadNames.PRED_NORMALS] = self.field_head_pred_normals(x) + outputs[FieldHeadNames.PRED_NORMALS] = math.safe_normalize(self.field_head_pred_normals(x)) h = density_embedding.view(-1, self.geo_feat_dim) diff --git a/nerfstudio/models/dreamfusion.py b/nerfstudio/models/dreamfusion.py index 3bfe1372d..494a5eaea 100644 --- a/nerfstudio/models/dreamfusion.py +++ b/nerfstudio/models/dreamfusion.py @@ -149,7 +149,6 @@ class DreamFusionModel(Model): # samplers self.sampler_uniform = UniformSampler(num_samples=self.num_samples, single_jitter=True) - self.sampler_pdf = PDFSampler(num_samples=self.num_samples // 2, single_jitter=True) # renderers self.renderer_rgb = RGBRenderer(background_color=colors.WHITE)