Pytorch Note52 灵活的数据读取介绍

本文介绍了PyTorch中灵活的数据读取方式,包括处理图片数据的两种常见情况:所有图片在一个文件夹内配以txt标签,以及不同类别图片存放在各自文件夹中。通过自定义Dataset和DataLoader,实现数据预处理和批处理功能,展示了如何根据需求调整collate_fn来处理batch数据。
摘要由CSDN通过智能技术生成

Pytorch Note52 灵活的数据读取介绍


全部笔记的汇总贴: Pytorch Note 快乐星球

图片数据一般有两种情况:

1、所有图片放在一个文件夹内,另外有一个txt文件显示标签。

2、不同类别的图片放在不同的文件夹内,文件夹就是图片的类别。

针对这两种不同的情况,数据集的准备也不相同,第一种情况可以自定义一个Dataset,第二种情况直接调用torchvision.datasets.ImageFolder来处理。下面分别进行说明:

灵活的数据读取

首先导入我们需要的函数

from torchvision.datasets import ImageFolder

文件中数据分布是这样的,每个文件夹中有三张图片

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

风信子的猫Redamancy

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值