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 的圖片 Tensor，其中每個顏色對應到給定的流量方向。形狀取決於輸入，為 (N, 3, H, W) 或 (3, H, W)。 回傳類型: img (Tensor) 使用 flow_to_image 的範例 光流：使用 RAFT 模型預測運動 光流：使用 RAFT 模型預測運動
