PyTorch加载数据集

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

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

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

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

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

相关推荐
熊猫钓鱼>_>5 小时前
鸿蒙ArkUI全手势操作实战指南:6大基础手势从原理到落地避坑
人工智能·深度学习·华为·架构·harmonyos·arkui·tapgesture
小小帅呀6 小时前
学习 VLA 第 2 天:深度学习基础
人工智能·深度学习·学习
工业机器视觉设计和实现7 小时前
opencv的模板匹配
opencv·计算机视觉
CoderIsArt8 小时前
基于Halcon的T恤印花缺陷检测程序
人工智能·数码相机·计算机视觉
xwz小王子8 小时前
Nature Sensors封面:清华“天眸芯”,类脑互补视觉范式重塑AI感知世界的方式
人工智能·深度学习·机器学习
飞奔的猫9 小时前
锈了儿—游戏素材用的在线图形处理工具
图像处理·人工智能·游戏美术
selia107810 小时前
AI手撕代码笔记
人工智能·笔记·深度学习
薛定e的猫咪10 小时前
(arXiv 2026)GLiBRL :可学习基函数的深度贝叶斯元强化学习 ----待补充
人工智能·深度学习·学习·算法·机器学习
躺柒11 小时前
读数据可视化13空间标量场(上)
人工智能·深度学习·信息可视化·数据可视化·空间·大数据分析
小王20411 小时前
Day 28:目标检测入门 — 两阶段 vs 单阶段
人工智能·目标检测·计算机视觉