tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

fix include_input for NeRFEncoding (#2642)

Browse Source
This commit is contained in:
blacksino 2023-12-01 07:07:26 +08:00 committed by GitHub
parent 3d8af87290
commit 747dd557f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nerfstudio/field_components/encodings.py
View File

@ -169,17 +169,18 @@ class NeRFEncoding(Encoding):
encoded_inputs = expected_sin(
torch.cat([scaled_inputs, scaled_inputs + torch.pi / 2.0], dim=-1), torch.cat(2 * [input_var], dim=-1)
)
if self.include_input:
encoded_inputs = torch.cat([encoded_inputs, in_tensor], dim=-1)
return encoded_inputs
def forward(
self, in_tensor: Float[Tensor, "*bs input_dim"], covs: Optional[Float[Tensor, "*bs input_dim input_dim"]] = None
) -> Float[Tensor, "*bs output_dim"]:
if self.tcnn_encoding is not None:
return self.tcnn_encoding(in_tensor)
return self.pytorch_fwd(in_tensor, covs)
encoded_inputs = self.tcnn_encoding(in_tensor)
else:
encoded_inputs = self.pytorch_fwd(in_tensor, covs)
if self.include_input:
encoded_inputs = torch.cat([encoded_inputs, in_tensor], dim=-1)
return encoded_inputs
class FFEncoding(Encoding):