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 Image，同時根據 mode 調整值範圍。

引數:

mode (PIL.Image 模式) –

輸入資料的顏色空間和畫素深度（可選）。如果 mode 為 None（預設值），則會假定輸入資料的一些屬性

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

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

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

• 如果輸入有 1 個通道，則 mode 由資料型別決定（例如 int、float、short）。