INNER CODE UNIT · Python

MVDepthLoss

nianticlabs/simplerecon · losses.py:79

class MVDepthLoss(nn.Module):
    def __init__(self, height, width):
        super().__init__()

        self.height = height
        self.width = width

        self.backproject = BackprojectDepth(self.height, self.width)
        self.project = Project3D()


    def get_valid_mask(
                    self,
                    cur_depth_b1hw,
                    src_depth_b1hw,
                    cur_invK_b44,
                    src_K_b44,
                    cur_world_T_cam_b44,

View source record →