import torch
from torch.utils import data
from PIL import Image
import numpy as np
from torchvision import transforms
from torch.utils.data import DataLoader
import torchvision
import matplotlib.pyplot as plt
import glob
导入数据的图片和标签
all_imgs_path = glob.glob(r"G:\Proiect\Pytorch1.6\pytorch进阶\dataset2\*.jpg")print("all_imgs_path[:5]:\t", all_imgs_path[:5])#print(len(weather_dataset))#print(weather_dataset[567:589])
species =["cloudy","rain","shine","sunrise"]
species_to_idx =dict((c, i)for i, c inenumerate(species))#print("species_to_idx:\t", species_to_idx)
idx_to_species =dict((v, k)for k, v in species_to_idx.items())print("idx_to_species:\t", idx_to_species)# print(next(iter(weather_dataset)))
all_labels=[]# 提取所有图片的标签for img in all_imgs_path:for i, c inenumerate(species):if c in img:
all_labels.append(i)#print(all_labels[:5])#print(all_labels[-5:])#print(all_labels)