PyTorch加载数据集

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

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

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

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

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

相关推荐
Jack___Xue25 分钟前
LLM知识随笔(一)--Transformer
人工智能·深度学习·transformer
罗兰Yolanda33 分钟前
影视后期全流程的核心软件及工作站配置方案推荐
计算机视觉·音视频
AI即插即用1 小时前
即插即用系列 | CVPR 2025 MK-UNet: 多核深度可分离卷积,重新定义轻量级医学图像分割
图像处理·人工智能·深度学习·神经网络·计算机视觉·视觉检测
发光的叮当猫1 小时前
什么是梯度
人工智能·深度学习
Struart_R1 小时前
VideoLLM相关论文(二)
计算机视觉·大语言模型·强化学习·多模态·r1
淡忘旧梦1 小时前
词错误率/WER算法讲解
人工智能·笔记·python·深度学习·算法
2501_936146041 小时前
柿子目标检测实战:YOLO11-HSFPN网络优化与性能分析
人工智能·目标检测·计算机视觉
_codemonster1 小时前
手语识别及翻译项目实战系列(一)环境准备
人工智能·python·计算机视觉
AAD555888991 小时前
【YOLO13项目实战】(5)镰刀目标检测与识别_C3k2_MBRConv3改进版
人工智能·目标检测·计算机视觉
wfeqhfxz25887822 小时前
柿子与桃子目标检测识别-YOLO11-seg-HGNetV2改进实现
人工智能·目标检测·计算机视觉