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