INNER CODE UNIT · Python

uv_coords

nianticlabs/simplerecon · losses.py:117

        uv_coords = 2 * uv_coords - 1

        src_depth_sampled_b1hw = F.grid_sample(
                                            input=src_depth_b1hw,
                                            grid=uv_coords,
                                            padding_mode='zeros',
                                            mode='nearest',
                                            align_corners=False,
                                        )

        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

View source record →