PyTorch加载数据集

PyTorch加载数据集主要涉及两个类------Dataset和Dataloader

Dataset类是提供一种方式去获取数据及其对应的真实的label值,Dataset类是需要我们自己写的。

实现的功能:一如何获取每一个数据及其label;二告诉我们总共有多少个数据

Dataloader类是为后面的网络提供不同的数据形式

举个例子:以垃圾分类来举例,其中垃圾就代表我们所要处理的数据。Dataset类的作用是将一堆垃圾进行分类,并在每一类中对垃圾进行0,1,2,......的编号,并获取具体的label值。Dataloader类的作用是将Dataset类编码之后的数据按照一定的大小(比如batch-size)进行打包

相关推荐
十铭忘11 分钟前
SAM2跟踪的理解19——位置编码
人工智能·深度学习·计算机视觉
张二娃同学15 分钟前
深度学习入门篇——Github的使用和项目的导入
人工智能·git·深度学习·开源·github
一个处女座的程序猿O(∩_∩)O18 分钟前
transformer模型:彻底改变AI格局的革命性架构
人工智能·深度学习·transformer
水龙吟啸41 分钟前
基于Orbbec-Gemini深度相机与SFM-2D to 3D重建算法、手部识别视觉算法、Unity运动控制的3D水果切割游戏
python·深度学习·神经网络·c#·游戏引擎·3d视觉·3d重建
白日做梦Q1 小时前
数据增强策略:不仅仅是旋转和翻转
人工智能·深度学习
重生之我要成为代码大佬2 小时前
深度学习1-安装pytorch(无独立显卡版本)
人工智能·pytorch·深度学习·机器学习
Lian_Ge_Blog2 小时前
微调方法学习总结(万字长文!)
人工智能·深度学习
EEPI2 小时前
【论文阅读】VGGT: Visual Geometry Grounded Transformer
论文阅读·深度学习·transformer
水月wwww2 小时前
【深度学习】循环神经网络实现文本预测生成
人工智能·rnn·深度学习·gru·lstm·循环神经网络·文本续写
zl_vslam3 小时前
SLAM中的非线性优-3D图优化之相对位姿Between Factor位姿图优化(十三)
人工智能·算法·计算机视觉·3d