深度学习pytorch代码:数据加载 Dataset() Dataloader

深度学习pytorch代码:数据加载 Dataset() Dataloader,第1张

Dataset(): 提供一种方式去获取数据及其label

功能:

  1. 如何获取每个数据及其label
  2. 告诉我们总共有多少数据

Dataloader(): 为后面的网络提供不同的数据形式(对数据进行打包)

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)
        self.img_path = os.listdir(self.path)






    def __getitem__(self, idex):
        img_name = self.img_path[idex]
        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)

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/568128.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-09
下一篇 2022-04-09

发表评论

登录后才能评论

评论列表(0条)

保存