Parameters for predictors with default values.
| 162 | |
| 163 | @dataclasses.dataclass |
| 164 | class PredictorParams: |
| 165 | """Parameters for predictors with default values.""" |
| 166 | |
| 167 | # Parameters for submodules. |
| 168 | initializer: InitializerParams = dataclasses.field(default_factory=InitializerParams) |
| 169 | monodepth: MonodepthParams = dataclasses.field(default_factory=MonodepthParams) |
| 170 | monodepth_adaptor: MonodepthAdaptorParams = dataclasses.field( |
| 171 | default_factory=MonodepthAdaptorParams |
| 172 | ) |
| 173 | gaussian_decoder: GaussianDecoderParams = dataclasses.field( |
| 174 | default_factory=GaussianDecoderParams |
| 175 | ) |
| 176 | # How to align depth map (only relevant for RGBGaussianPredictor). |
| 177 | depth_alignment: AlignmentParams = dataclasses.field(default_factory=AlignmentParams) |
| 178 | |
| 179 | # Selectively reduce learning rate for different properties. |
| 180 | delta_factor: DeltaFactor = dataclasses.field(default_factory=DeltaFactor) |
| 181 | # The maximum scale of Gaussians relative to initial scale. |
| 182 | max_scale: float = 10.0 |
| 183 | # The minimum scale of Gaussians relative to initial scale. |
| 184 | min_scale: float = 0.0 |
| 185 | # Which normalization to use in prediction head. |
| 186 | norm_type: NormLayerName = "group_norm" |
| 187 | # How many groups to use for group normalization. |
| 188 | norm_num_groups: int = 8 |
| 189 | # Whether to use predicted mean to sample triplane features. |
| 190 | use_predicted_mean: bool = False |
| 191 | # Which activation function to use for colors / opacities. |
| 192 | color_activation_type: math_utils.ActivationType = "sigmoid" |
| 193 | opacity_activation_type: math_utils.ActivationType = "sigmoid" |
| 194 | # Colorspace of the renderer ("linearRGB" or "sRGB"). |
| 195 | color_space: ColorSpace = "linearRGB" |
| 196 | # A small value to avoid ill-conditioned splats |
| 197 | low_pass_filter_eps: float = 1e-2 |
| 198 | # How many layer of depth does monodepth model predict. |
| 199 | num_monodepth_layers: int = 2 |
| 200 | # Whether to sort the monodepth output (for two layer monodepth). |
| 201 | sorting_monodepth: bool = False |
| 202 | # Whether to account the z offsets for estimating base scale. |
| 203 | base_scale_on_predicted_mean: bool = True |