pytorch——Dataset类代码实战

最新推荐文章于 2024-02-13 19:49:50 发布

cfqp

最新推荐文章于 2024-02-13 19:49:50 发布

阅读量251

点赞数

分类专栏： pytorch python 文章标签： pytorch 深度学习神经网络人工智能图像处理 Powered by 金山文档

本文链接：https://blog.csdn.net/cfqp123456/article/details/128605538

版权

pytorch 同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

python

3 篇文章 0 订阅

订阅专栏

该视频教程详细介绍了如何使用PyTorch加载ants-bees二分类数据集。首先从给出的链接下载数据集，然后通过自定义的MyDataSet类来组织和访问图像数据，最后将ants和bees两个数据集合并为一个数据集进行使用。

摘要由CSDN通过智能技术生成

视频教程：P5. PyTorch加载数据初认识_哔哩哔哩_bilibili

数据集：ants-bees二分类数据集

数据集下载地址： https://pan.baidu.com/s/1jZoTmoFzaTLWh4lKBHVbEA 密码: 5suq

代码如下（建议先跟着UＰ主边听边做，最后自己再敲一遍）

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


class MyDataSet(Dataset):
    def __init__(self, dir_root, label):
        self.dir_root = dir_root
        self.label = label
        # 获取数据根目录
        path = os.path.join(self.dir_root, self.label)
        self.image_name_list = os.listdir(path)

    def __getitem__(self, index):
        image_name = self.image_name_list[index]
        image_path = os.path.join(self.dir_root, self.label, image_name)
        image = Image.open(image_path)
        label = self.label
        return image, label

    def __len__(self):
        return len(self.image_name_list)


data_ants = MyDataSet("data_set/train", "ants")
#image, label = data_ants[0]

data_bees = MyDataSet("data_set/train","bees")

data=data_ants+data_bees　　＃两个数据集合并