PyTorch加载数据集

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

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

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

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

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

相关推荐
点PY16 分钟前
体绘制传递函数优化论文精读(1)
图像处理
毕胜客源码19 分钟前
卷积神经网络的农作物识别系统(有技术文档)深度学习 图像识别 卷积神经网络 Django python 人工智能
人工智能·python·深度学习·cnn·django
小鱼~~1 小时前
GRU模型简介
人工智能·深度学习
小鱼~~1 小时前
DataLoader简介
人工智能·深度学习
AI木马人1 小时前
7.计算机视觉:让AI拥有一双“火眼金睛”
人工智能·计算机视觉
多年小白1 小时前
谷歌第八代 TPU 来了:性能提升 124%
网络·人工智能·科技·深度学习·ai
西门大盗1 小时前
pycharm自动进行python 测试(python test)
ide·python·pycharm
六bring个六2 小时前
opencv初始环境搭建
opencv·计算机视觉
懷淰メ2 小时前
【AI加持】基于PyQt+YOLO+DeepSeek的布匹缺陷检测系统(详细介绍)
yolo·目标检测·计算机视觉·pyqt·缺陷检测·布匹·布匹缺陷
AI木马人2 小时前
1.【AI系统架构设计】如何设计一个高效、安全的人性化AI工具系统?(从0到1完整方案)
人工智能·深度学习·神经网络·计算机视觉·自然语言处理