小编Ham*_*man的帖子

img 应该是 PIL Image。得到 <class 'torch.Tensor'>

我正在尝试遍历加载程序以检查它是否正常工作,但是给出了以下错误:

TypeError: img should be PIL Image. Got <class 'torch.Tensor'>

我试着将两者transforms.ToTensor()transforms.ToPILImage()和它给我一个错误,要求相反。即,使用ToPILImage(),它将要求张量,反之亦然。

# Imports here
%matplotlib inline
import matplotlib.pyplot as plt
from torch import nn, optim
import torch.nn.functional as F
import torch
from torchvision import transforms, datasets, models
import seaborn as sns
import pandas as pd
import numpy as np

data_dir = 'flowers'
train_dir = data_dir + '/train'
valid_dir = data_dir + '/valid'
test_dir = data_dir + '/test'

#Creating transform for training set
train_transforms = …
Run Code Online (Sandbox Code Playgroud)

python pytorch

8
推荐指数
2
解决办法
2万
查看次数

标签 统计

python ×1

pytorch ×1