flow_to_image

torchvision.utils.flow_to_image(flow: torch.Tensor) → torch.Tensor

Converts a flow to an RGB image.

Parameters

flow (Tensor) – Flow of shape (N, 2, H, W) or (2, H, W) and dtype torch.float.

Returns

Image Tensor of dtype uint8 where each color corresponds

to a given flow direction. Shape is (N, 3, H, W) or (3, H, W) depending on the input.

Return type

img (Tensor)

Examples using flow_to_image:

Optical Flow: Predicting movement with the RAFT model