Shortcuts

to_pil_image

torchvision.transforms.functional.to_pil_image(pic, mode=None)[source]

将张量或ndarray转换为PIL图像。此函数不支持torchscript。

有关更多详细信息,请参见ToPILImage

Parameters:
  • pic (Tensornumpy.ndarray) – 要转换为PIL图像的图像。

  • mode (PIL.Image mode) – 输入数据的颜色空间和像素深度(可选)。

Returns:

图像已转换为PIL图像。

Return type:

PIL 图像

使用 to_pil_image 的示例:

光流:使用RAFT模型预测运动

Optical Flow: Predicting movement with the RAFT model

将口罩重新利用为边界框

Repurposing masks into bounding boxes

可视化工具

Visualization utilities