pytorch深度学习快速入门-编程知识

pytorch深度学习快速入门

from PIL import Image
img_path="E:\\code\\learn_pytorch\\dataset\\train\\ants\\0013035.jpg"
img = Image.open(img_path)

在这里插入图片描述

显示图片

获取很多图
在这里插入图片描述

    def __init__(self,root_dir,label_dir):self.root_dir = root_dirself.label_dir = label_dirself.path = os.path.join(self.root_dir,self.label_dir)self.img_path = os.listdir(self.path) #获取图片列表

在这里插入图片描述

import os
from torch.utils.data import Dataset
from PIL import Imageclass MyData(Dataset):##初始化类，根据这个类去创建特定的实例去运行的函数，这个函数中一般写的是全局变量def __init__(self, root_dir, label_dir):self.root_dir = root_dirself.label_dir = label_dirself.path = os.path.join(self.root_dir,self.label_dir)self.img_path = os.listdir(self.path) #获取图片名称列表#获取其中每个图片和标签def __getitem__(self, idx):#图像名列表img_name = self.img_path[idx]#每一个图像具体的路径img_item_path = os.path.join(self.root_dir,self.label_dir,img_name)img = Image.open(img_item_path) #读取文件label = self.label_dir#返回图像和标签return img,labeldef __len__(self):return len(self.img_path)root_dir="dataset/train"
label_ants="ants"
label_bees="bees"
dataset_ants=MyData(root_dir,label_ants)
dataset_bees=MyData(root_dir,label_bees)
train_dataset = dataset_ants+dataset_bees