From 6cb10ae682030b2d66bb45049b3d765002e24948 Mon Sep 17 00:00:00 2001 From: Riley Date: Tue, 11 Jul 2023 15:56:12 -0700 Subject: [PATCH] Adding input handling for RFFEncoding's get_out_dim(self) (#2201) fixed RFFEncoding's get_out_dim(self) to incude dims of input if include_input --- nerfstudio/field_components/encodings.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nerfstudio/field_components/encodings.py b/nerfstudio/field_components/encodings.py index 6e0158654..7145ae599 100644 --- a/nerfstudio/field_components/encodings.py +++ b/nerfstudio/field_components/encodings.py @@ -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,