Method forward_gaussians

model/encoder_model.py:62–66

(self, frames, depths, cond_times=None)

        return self
    
    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 forward(self, data, step_ratio=0.0):
        # data: [B, 2, C, H, W]