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 模型預測運動
