Method forward_gaussians

model/splat_model.py:63–67

(self, frames, depths, cond_times=None)

        return super().load_state_dict(filtered_state_dict, strict=strict)
        
    def forward_gaussians(self, frames, depths, cond_times=None):
        # frames: [B, V, C, H, W]
        # return: gaussians: [B, N, D]
        decoder_out = self.model(frames, depths, cond_times=cond_times)
        return decoder_out
    
    def compute_losses(self, input_frames, target_depth, supv_masks, render_pkg):
        output_frames = render_pkg["render"]  # [B, V, C, H, W]