PyTorch加载数据集

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

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

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

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

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

相关推荐
咚咚王者8 小时前
人工智能之核心技术 深度学习 第六章 生成对抗网络(GAN)
人工智能·深度学习·生成对抗网络
子夜江寒8 小时前
基于dlib与OpenCV的人脸检测与特征点标定技术实践
人工智能·opencv·计算机视觉
IRevers8 小时前
RF-DETR:第一个在COCO上突破60AP的DETR(含检测和分割推理)
图像处理·人工智能·python·深度学习·目标检测·计算机视觉
是小蟹呀^8 小时前
卷积神经网络(CNN):池化操作
人工智能·深度学习·神经网络·cnn
Pyeako8 小时前
opencv计算机视觉--光流估计&视频读取方法
python·opencv·计算机视觉·pycharm·角点检测·光流估计·视频读取方法
何中应8 小时前
PyCharm报`Invalid Python SDK`错误
ide·python·pycharm
薛定谔的猫19828 小时前
十四、基于 BERT 的微博评论情感分析模型训练实践
人工智能·深度学习·bert
学步_技术8 小时前
食品计算-Multimodal Food Learning
人工智能·深度学习·计算机视觉·语言模型
中國龍在廣州10 小时前
AI时代“新BAT”正在崛起
大数据·人工智能·深度学习·重构·机器人
丝瓜蛋汤10 小时前
unsloth 部署(简单易上手版本)
人工智能·深度学习