flow_to_image¶
-
torchvision.utils.flow_to_image(flow: torch.Tensor) → torch.Tensor[source]¶ 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
Optical Flow: Predicting movement with the RAFT model
