tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Adding input handling for RFFEncoding's get_out_dim(self) (#2201)

Browse Source 
fixed RFFEncoding's get_out_dim(self) to incude dims of input if include_input
This commit is contained in:
Riley 2023-07-11 15:56:12 -07:00 committed by GitHub
parent 70adb01b4c
commit 6cb10ae682
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nerfstudio/field_components/encodings.py
View File

@ -208,7 +208,12 @@ class RFFEncoding(Encoding):
self.include_input = include_input
def get_out_dim(self) -> int:
return self.num_frequencies * 2
out_dim = self.num_frequencies * 2
if self.include_input:
if self.in_dim is None:
raise ValueError("Input dimension has not been set")
out_dim += self.in_dim
return out_dim
def forward(
self,