我正在尝试遍历加载程序以检查它是否正常工作,但是给出了以下错误:
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)