flow_to_image¶ torchvision.utils.flow_to_image(flow: Tensor) → Tensor[源代码]¶ 将流转换为 RGB 图像。 参数: flow (Tensor) – 形状为 (N, 2, H, W) 或 (2, H, W) 且 dtype 为 torch.float 的流。 返回值: dtype 为 uint8 的图像张量,其中每种颜色对应给定的流方向。形状为 (N, 3, H, W) 或 (3, H, W),具体取决于输入。 返回类型: img (Tensor) 使用 flow_to_image 的示例 光流:使用 RAFT 模型预测运动 光流:使用 RAFT 模型预测运动