(self, output)
| 587 | return encoder_output |
| 588 | |
| 589 | def upsampling(self, output): |
| 590 | input_views = output.shape[1] |
| 591 | if self.decoder_ratio > 0: |
| 592 | output = output.reshape(output.shape[0], -1, output.shape[-1]) # [B, V, N, D] -> [B, V*N, D] |
| 593 | output = self.gaussian_upsampler(output) # [B, V*N, D] |
| 594 | output = rearrange(output, 'b (v n) d -> b v n d', v=input_views) |
| 595 | return output |
| 596 | |
| 597 | def forward_condition(self, frames, depths): |
| 598 | # [B, V, C, H, W] |