ToTensor¶
- class torchvision.transforms.ToTensor[source]¶
將 PIL Image 或 ndarray 轉換為 tensor 並相應地縮放值。
此變換不支援 torchscript。
將範圍在 [0, 255] 的 PIL Image 或 numpy.ndarray (H x W x C) 轉換為形狀為 (C x H x W)、範圍在 [0.0, 1.0] 的 torch.FloatTensor，前提是 PIL Image 屬於以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1)，或者 numpy.ndarray 的 dtype 為 np.uint8。
在其他情況下，返回的 tensors 不進行縮放。
注意
由於輸入影像被縮放至 [0.0, 1.0] 範圍，因此在對目標影像 mask 進行變換時，不應使用此變換。請參閱參考資料，瞭解如何實現影像 mask 的變換。
