(self, frames, depths, cond_times=None)
| 509 | return output |
| 510 | |
| 511 | def forward(self, frames, depths, cond_times=None): |
| 512 | batch_size, input_views = frames.shape[0], frames.shape[1] |
| 513 | encoder_output, target_depth, mask = self.forward_encoder(frames, depths, cond_times) # [B, V, N, D] |
| 514 | output = encoder_output |
| 515 | num_features = output.shape[2] |
| 516 | |
| 517 | output = self.upsampling(output) # [B, V, N, D] |
| 518 | |
| 519 | # Decoding |
| 520 | pred_gs, prior_params = self.predictor(output) |
| 521 | |
| 522 | return {'pred_gs': pred_gs, 'gt_depth': target_depth, 'gt_depth_mask': mask}, prior_params |
| 523 | |
| 524 | class SplatPredictor(nn.Module): |
| 525 | def __init__(self, opt: Options, **model_kwargs): |
nothing calls this directly
no test coverage detected