ToPILImage

class torchvision.transforms.ToPILImage(mode=None)

將張量或 ndarray 轉換為 PIL 影像

此轉換不支援 torchscript。

將形狀為 C x H x W 的 torch.*Tensor 或形狀為 H x W x C 的 numpy ndarray 轉換為 PIL 影像，同時根據 mode 調整數值範圍。

參數:

mode (PIL.Image 模式) –

輸入資料的色彩空間和像素深度 (可選)。如果 mode 為 None (預設)，則會對輸入資料做一些假設

• 如果輸入有 4 個通道，則 mode 假設為 RGBA。

• 如果輸入有 3 個通道，則 mode 假設為 RGB。

• 如果輸入有 2 個通道，則 mode 假設為 LA。

• 如果輸入有 1 個通道，則 mode 由資料類型決定 (即 int, float, short)。