INNER CODE UNIT · Python

forward

nianticlabs/simplerecon · losses.py:175

    def forward(
                self,
                depth_pred_b1hw,
                cur_depth_b1hw,
                src_depth_bk1hw,
                cur_invK_b44,
                src_K_bk44,
                cur_world_T_cam_b44,
                src_cam_T_world_bk44,
            ):

        src_to_iterate = [
                            torch.unbind(src_depth_bk1hw, dim=1),
                            torch.unbind(src_K_bk44, dim=1),
                            torch.unbind(src_cam_T_world_bk44, dim=1)
                        ]

        num_src_frames = src_depth_bk1hw.shape[1]

View source record →