python自定义数据集并展示

最新推荐文章于 2024-10-02 17:22:34 发布

yeahamen

最新推荐文章于 2024-10-02 17:22:34 发布

阅读量833

点赞数 1

文章标签： python 深度学习人工智能

本文链接：https://blog.csdn.net/amen_yanwang/article/details/130497443

版权

该文介绍了如何在Python中利用PyTorch框架创建自定义的彩色图像数据集。通过定义一个继承自`torch.utils.data.Dataset`的类，实现了读取图像和标签、数据预处理等功能。此外，还展示了如何使用DataLoader进行批量处理，并对图像进行可视化展示。

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

首先，我自己创建的彩色图像数据集是这样的：

标签是这样的：

#本文引荐了文章：http://t.csdn.cn/gkVNC；并作了注释与修改
#导入库
import os
import torch
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from torchvision.io import read_image
from torch.utils.data import Dataset
from torch.utils.data import DataLoader
from torchvision import transforms

#创建自定义数据集类
class Custom_Dataset(Dataset):
    #函数，设置图像集路径索引、图像标签文件读取
    def __init__(self, img_dir, img_label_dir, transform=None):
        super().__init__()
        self.img_dir = img_dir
        self.img_labels = pd.read_csv(img_label_dir)
        self.transform