pytorch使用入门1

一、pytorch环境配置：https://www.bilibili.com/video/BV1hE411t7RN?p=1

二、jupyter启动问题：

在终端输入：jupyter-notebook后，在使用过程中出现：Unhandle error

此时jupyter是不能用的，在终端输入pip list发现在pytorch环境下未安装pywin32，安装一下就好pip install pywin32==227(与base环境中一致就行)

 三、如何读取图片：

数据存放路径如下：ants为文件夹名label，该类数据存放在文件夹内

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

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)  //组合为lable文件夹的相对路径
        self.img_path = os.listdir(self.path)  //文件夹内所有图片相对路径列表

    def __getitem__(self, idx): //获得idx编号的图片和其label
        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 = "hymenoptera_data/train"

ants_label_dir = "ants"
bees_label_dir = "bees"

ants_dataset = MyData(root_dir,ants_label_dir)//实例调用使用根目录和标签即可
bees_dataset = MyData(root_dir,bees_label_dir)

train_dataset = ants_dataset + bees_dataset//把两个数据集拼合

//调用

  img,label = train_dataset[0]
  img.show()

pytorch使用入门1

原文：https://www.cnblogs.com/zyf-f/p/14408675.html