flow_to_image
-
torchvision.utils.flow_to_image(flow: Tensor) → Tensor[原始碼]
將光流轉換為 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 的示例
