2f49884ff8169422cbb711e18e94464efc59789c,torchgeometry/core/pinhole.py,,normalize_pixel_coordinates,#Any#Any#Any#,603

Before Change

    Return:
        torch.Tensor: the nornmalized pixel coordinates.
    
    if len(pixel_coordinates.shape) != 4 and pixel_coordinates.shape[-1] != 2:
        raise ValueError("Input pixel_coordinates must be of shape BxHxWx2. "
                         "Got {}".format(pixel_coordinates.shape))

    // unpack pixel coordinates
    u_coord, v_coord = torch.chunk(pixel_coordinates, dim=-1, chunks=2)

After Change

                         "Got {}".format(pixel_coordinates.shape))

    // compute normalization factor
    hw: torch.Tensor = torch.stack([
        torch.tensor(width), torch.tensor(height)]
    ).to(pixel_coordinates.device).to(pixel_coordinates.dtype)

    factor: torch.Tensor = torch.tensor(2.) / (hw - torch.tensor(1.))

    // normalize coordinates and return

Instances