快捷方式

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 模型预测运动

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源