比较与torchvision.transforms.ToPILImage的差异
torchvision.transforms.ToPILImage
class torchvision.transforms.ToPILImage(
mode=None
)
mindspore.dataset.vision.ToPIL
class mindspore.dataset.vision.ToPIL
差异对比
PyTorch:将torch中的Tensor或Numpy数组转换为PIL类型的图像。输入可以是<C, H, W> 格式的torch Tensor,或者<H, W, C> 格式的Numpy数组。
MindSpore:将<H, W, C>格式的Numpy数组(如解码后的图像)转换为PIL图像,不支持指定输入图像的颜色空间。
分类 |
子类 |
PyTorch |
MindSpore |
差异 |
---|---|---|---|---|
参数 |
参数1 |
mode |
- |
输入原始数据的颜色空间 |
代码示例
import numpy as np
import torch as T
from torchvision.transforms import ToPILImage
import mindspore.dataset.vision as vision
# In MindSpore, ToPIL transform the numpy.ndarray to PIL Image.
image = np.random.random((64,64))
img = vision.ToPIL()(image)
img.show()
# Out:
# window of PIL image
# In torch, ToPILImage transforms the input to PIL Image.
image = T.randn((64, 64))
img = ToPILImage()(image)
img.show()
# Out:
# window of PIL image