INNER CODE UNIT · Python

valid_mask_b1hw

nianticlabs/simplerecon · losses.py:127

        valid_mask_b1hw = proj_src_depths_b1hw < 1.05 * src_depth_sampled_b1hw
        valid_mask_b1hw = torch.logical_and(valid_mask_b1hw, 
                                                    proj_src_depths_b1hw > 0)
        valid_mask_b1hw = torch.logical_and(valid_mask_b1hw, 
                                                    src_depth_sampled_b1hw > 0)

        return valid_mask_b1hw, src_depth_sampled_b1hw


    def get_error_for_pair(self,
                           depth_pred_b1hw,
                           cur_depth_b1hw,
                           src_depth_b1hw,
                           cur_invK_b44,
                           src_K_b44,
                           cur_world_T_cam_b44,
                           src_cam_T_world_b44):

View source record →