pytorch里面报错raise TypeError('tensor is not a torch image.')
发布日期:2021-07-01 04:36:56 浏览次数:102 分类:技术文章

本文共 648 字,大约阅读时间需要 2 分钟。

报错如下:

File “/home/muli/anaconda3/lib/python3.5/site-packages/torchvision/transforms/functional.py”, line 158, in normalize
raise TypeError(‘tensor is not a torch image.’)
TypeError: tensor is not a torch image.

解决办法

transforms.ToTensor()操作要在transforms.Normalize(mean=[0.56391764, 0.43714827, 0.4107524], std=[0.22986116, 0.21178758, 0.20076773])操作的前面

trainTransform = transforms.Compose([         transforms.RandomResizedCrop(224), # 随机裁剪,         transforms.RandomHorizontalFlip(), # 随机水平翻转         transforms.ToTensor(),         transforms.Normalize(mean=[0.56391764, 0.43714827, 0.4107524], std=[0.22986116, 0.21178758, 0.20076773]) # 正则化    ])

转载地址:https://mymuli.blog.csdn.net/article/details/100836968 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:AP和mAP 计算:sklearn.metrics.average_precision_score()
下一篇:pytorch打印自定义网络的每层的名称

发表评论

最新留言

留言是一种美德,欢迎回访!
[***.207.175.100]2024年04月22日 22时54分07秒