pytorch学习笔记(2)-使用自定义txt文件读取数据

最新推荐文章于 2025-05-10 21:25:53 发布

_rookie_coder

最新推荐文章于 2025-05-10 21:25:53 发布

阅读量6.5k

点赞数 1

分类专栏： pytorch 文章标签： pytorch 自定义数据数据载入

本文链接：https://blog.youkuaiyun.com/qq_20622615/article/details/83185289

版权

本文详细介绍了如何在PyTorch中使用自定义的TXT文件作为数据源，进行数据加载和预处理，适用于需要处理文本数据的机器学习任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
import torch
from PIL import Image
from torch.utils.data import Dataset
from torchvision import datasets, models, transforms
class dataLoader(Dataset):
    def __init__(self, path, listName, dataset = '', data_transforms = None, loader = None):
        self.path = path
        self.listName = listName
        self.images = [os.path.join(self.path, line.strip().split()[0]) for line in open(self.listName)]
        self.labels = [int(line.strip().split()[1]) for line in open(self.listName)]
        self.data_transforms = data_transforms
        self.dataset = dataset
        if loader:
            self.loader = loader
        else:
            self.loader = self.default_loader
    def default_loader(self, imageName):
        try:
            image = Image.open(imageName)
            return image.convert('RGB')
        except:
            print("Cannot read image", path)
    def __len__(self):