Create base scale for the gaussians.
(disparity: torch.Tensor, disparity_scale_factor: float)
| 272 | |
| 273 | |
| 274 | def _create_base_scale(disparity: torch.Tensor, disparity_scale_factor: float) -> torch.Tensor: |
| 275 | """Create base scale for the gaussians.""" |
| 276 | inverse_disparity = torch.ones_like(disparity) / disparity |
| 277 | base_scales = inverse_disparity * disparity_scale_factor |
| 278 | return base_scales |
| 279 | |
| 280 | |
| 281 | def _rescale_depth( |