tfg.rendering.camera.perspective.random_rays

Sample rays at random pixel location from a perspective camera.

tfg.rendering.camera.perspective.random_rays(
    focal: tf.Tensor,
    principal_point: tf.Tensor,
    height: int,
    width: int,
    n_rays: int,
    margin: int = 0,
    name: str = 'random_rays'
) -> Tuple[tf.Tensor, tf.Tensor]

focal A tensor of shape [A1, ..., An, 2] where the last dimension contains the fx and fy focal length values.
principal_point A tensor of shape [A1, ..., An, 2] where the last dimension contains the cx and cy principal point values.
height The height of the image plane in pixels
width The width of the image plane in pixels.
n_rays The number M of rays to sample.
margin The margin around the borders of the image.
name A name for this op that defaults to "random_rays".

A tensor of shape [A1, ..., An, M, 3] with the ray directions and a tensor of shape [A1, ..., An, M, 2] with the pixel x, y locations.