自定义数据集Dataset

最新推荐文章于 2024-10-15 07:20:54 发布

ai-ai360

最新推荐文章于 2024-10-15 07:20:54 发布

阅读量1.1k

点赞数 1

分类专栏： python 文章标签： pytorch dataset

本文链接：https://blog.csdn.net/q_xiami123/article/details/116600976

版权

本文介绍了如何在PyTorch中创建自定义数据集，通过修改__getitem__方法实现按需访问数据，并展示了将数据读取为(x,y)格式的步骤，包括使用transforms处理图像数据，以及数据集的划分过程。" 124974101,12654538,PostgreSQL与MySQL：全面解析两者差异,"['数据库', 'PostgreSQL', 'MySQL', 'SQL']

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

模板

from torch.utils.data import DataLoader, Dataset
class CatsAndDogsDataset(Dataset):
    def __init__(self, csv_file, root_dir, transform=None):
        self.data = pd.read_csv(csv_file)
        self.root_dir = root_dir
        self.transform = transform
    def __len__(self