pytorch深度学习快速入门

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

显示图片

获取很多图

    def __init__(self,root_dir,label_dir):
        self.root_dir = root_dir
        self.label_dir = label_dir
        self.path = os.path.join(self.root_dir,self.label_dir)
        self.img_path = os.listdir(self.path) #获取图片列表
1
2
3
4
5

import os
from torch.utils.data import Dataset
from PIL import Image

class MyData(Dataset):
    ##初始化类，根据这个类去创建特定的实例去运行的函数，这个函数中一般写的是全局变量
    def __init__(self, root_dir, label_dir):
        self.root_dir = root_dir
        self.label_dir = label_dir
        self.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,label

    def __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
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

Tensorboard的使用