TypeError: ToTensor() takes no arguments

anzrs

已于 2022-03-19 16:02:19 修改

阅读量800

点赞数 2

文章标签： python pycharm 机器学习 list pytorch

于 2022-03-19 15:59:16 首次发布

本文链接：https://blog.csdn.net/weixin_45193103/article/details/123596512

版权

代码如下：

from torchvision import transforms
from PIL import Image
img_path = "dataset/train/ants/0013035.jpg"
img = Image.open(img_path)
#tensor_trans = transforms.ToTensor()
#tensor_img = tensor_trans(img)
tensor_img = transforms.ToTensor(img)
print(tensor_img)
#TypeError: ToTensor() takes no arguments

原因，需要将transforms.ToTensor()传给一个变量再来使用

相当于 tool = transforms.ToTensor()

然后 result = tool(raw)

修改后的代码如下：

from torchvision import transforms
from PIL import Image
img_path = "dataset/train/ants/0013035.jpg"
img = Image.open(img_path)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img)
#TypeError: ToTensor() takes no arguments

输出的结果是img的向量表示

tensor([[[0.3137, 0.3137, 0.3137,  ..., 0.3176, 0.3098, 0.2980],
         [0.3176, 0.3176, 0.3176,  ..., 0.3176, 0.3098, 0.2980],
         [0.3216, 0.3216, 0.3216,  ..., 0.3137, 0.3098, 0.3020],
         ...,
         [0.3412, 0.3412, 0.3373,  ..., 0.1725, 0.3725, 0.3529],
         [0.3412, 0.3412, 0.3373,  ..., 0.3294, 0.3529, 0.3294],
         [0.3412, 0.3412, 0.3373,  ..., 0.3098, 0.3059, 0.3294]],

        [[0.5922, 0.5922, 0.5922,  ..., 0.5961, 0.5882, 0.5765],
         [0.5961, 0.5961, 0.5961,  ..., 0.5961, 0.5882, 0.5765],
         [0.6000, 0.6000, 0.6000,  ..., 0.5922, 0.5882, 0.5804],
         ...,
         [0.6275, 0.6275, 0.6235,  ..., 0.3608, 0.6196, 0.6157],
         [0.6275, 0.6275, 0.6235,  ..., 0.5765, 0.6275, 0.5961],
         [0.6275, 0.6275, 0.6235,  ..., 0.6275, 0.6235, 0.6314]],

        [[0.9137, 0.9137, 0.9137,  ..., 0.9176, 0.9098, 0.8980],
         [0.9176, 0.9176, 0.9176,  ..., 0.9176, 0.9098, 0.8980],
         [0.9216, 0.9216, 0.9216,  ..., 0.9137, 0.9098, 0.9020],
         ...,
         [0.9294, 0.9294, 0.9255,  ..., 0.5529, 0.9216, 0.8941],
         [0.9294, 0.9294, 0.9255,  ..., 0.8863, 1.0000, 0.9137],
         [0.9294, 0.9294, 0.9255,  ..., 0.9490, 0.9804, 0.9137]]])

进程已结束，退出代码为 0