在notebook上正常,在pycharm报错
import os
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
from torchvision import transforms,utils
from torchvision import datasets
import torch
from matplotlib import pyplot as plt
my_trans = transforms.Compose([
transforms.RandomResizedCrop(224),
transforms.RandomHorizontalFlip(),
transforms.ToTensor()
])
train_data = datasets.ImageFolder('E:/datasets/hymenoptera_data',transform=my_trans)
train_loader =torch.utils.data.DataLoader(train_data,batch_size=8,shuffle=True)
for i_batch,img in enumerate(train_loader):
if i_batch ==0:
print(img[1])
fig = plt.figure()
grid = utils.make_grid(img[0])
plt.imshow(grid.numpy().transpose((1,2,0)))
plt.show()
utils.save_image(grid,'test01.png')
break