Pytorch中使用transforms.ToTensor和transforms.ToPILImage的numpy.ndarray与Tensor和PILImage的转换举例
作为刚接触Pytorch的新人,在刚开始时候处理三种数据tensor,numpy,PIL有点弄晕了,而且中间的转换过程需要注意类型,仅此记录下来,如果能有幸帮到大家那更好了。进入正题,我们首先看看numpytotensor的情况:importnumpyasnpfromtorchvisionimporttransformsasTx=np.ones((128,228,1))#这里numpy维度是HWC