cc.日常

yeah,cc.

已于 2022-02-16 22:05:44 修改

阅读量1.6k

点赞数

分类专栏： cc. 文章标签：机器学习

于 2022-02-16 22:05:10 首次发布

本文链接：https://blog.csdn.net/qq_43641338/article/details/122967714

版权

cc. 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

这段代码展示了如何利用PyTorch的`torch.utils.data.Dataset`创建自定义数据集，用于加载和处理`dataset/train`目录下`ants`和`bees`两个子目录的图像数据。数据集被组合成`train_dataset`，其长度为245，可以用于训练分类模型。示例中展示了如何获取第124个样本并显示图像。

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

可以说土堆真的保姆级

学习读取数据集中的各类数据

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, 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"
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

python console

C:\ProgramData\Anaconda3\envs\pytorch\python.exe "C:\D\ccpython\PyCharm Community Edition 2021.3.1\plugins\python-ce\helpers\pydev\pydevconsole.py" --mode=client --port=53938
import sys; print('Python %s on %s' % (sys.version, sys.platform))
sys.path.extend(['C:\\Users\\26940\\Desktop\\learn_pytorchc', 'C:/Users/26940/Desktop/learn_pytorchc'])
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)]
Type 'copyright', 'credits' or 'license' for more information
IPython 6.1.0 -- An enhanced Interactive Python. Type '?' for help.
PyDev console: using IPython 6.1.0
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32
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, 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"
ants_label_dir = "ants"
ants_dataset = MyData(root_dir,ants_label_dir)
root_dir = "dataset/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
len(train_dataset)
Out[7]: 245
img,label=train_dataset[124]
img.show()

蜜蜂蚂蚁数据集：

https://download.pytorch.org/tutorial/hymenoptera_data.ziphttps://download.pytorch.org/tutorial/hymenoptera_data.zip