Skip to content

Commit 0bd8741

Browse files
committed
added sphere initialization forward function to the model.
1 parent 7fb65d4 commit 0bd8741

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

model.py

+7
Original file line numberDiff line numberDiff line change
@@ -712,6 +712,13 @@ def styles_and_noise_forward(self, styles, inject_index=None, truncation=1,
712712

713713
return styles
714714

715+
def init_forward(self, styles, cam_poses, focals, near=0.88, far=1.12):
716+
latent = self.styles_and_noise_forward(styles)
717+
718+
sdf, target_values = self.renderer.mlp_init_pass(cam_poses, focals, near, far, styles=latent[0])
719+
720+
return sdf, target_values
721+
715722
def forward(self, styles, cam_poses, focals, near=0.88, far=1.12, return_latents=False,
716723
inject_index=None, truncation=1, truncation_latent=None,
717724
input_is_latent=False, noise=None, randomize_noise=True,

0 commit comments

Comments
 (0)