news 2026/9/21 15:56:26

Day42Dataset和Dataloader

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Day42Dataset和Dataloader

1.Dataset类:“存数据的容器”

你可以把它理解成一个数据盒子,里面装着你的数据集(比如图片、标签)。要让这个 “盒子” 能用,得给它加两个 “功能按钮”(Python 的特殊方法):

  • __getitem__(self, idx):按索引idx取数据(比如dataset[0]就能拿到第 1 个样本)。
  • __len__(self):返回数据集的总样本数(比如len(dataset)知道一共有多少数据)。

2.DataLoader类:“给模型端菜的服务员”

Dataset是 “装菜的盘子”,DataLoader就是 “把菜分成小份、端给模型吃” 的人。它的作用是:

  • Dataset里的数据分成批次(比如一次给模型喂 32 个样本,而不是全塞进去)。
  • 支持打乱数据(避免模型学 “顺序” 而不是 “规律”)。
  • 支持多线程加载(加快数据读取速度)。

3. MNIST 手写数据集

这是一个 “手写数字图片集”,里面是 0-9 的手写数字(每张图是 28x28 的黑白色),是深度学习入门常用的 “练习数据集”。

作业:获取 CIFAR 数据集的一张图片

import torch from torchvision import datasets, transforms import matplotlib.pyplot as plt # 1. 加载CIFAR数据集(自动下载到本地) transform = transforms.ToTensor() # 把图片转成PyTorch能处理的格式 cifar_dataset = datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) # 2. 取第1张图和它的标签 img, label = cifar_dataset[0] # 用Dataset的__getitem__取数据 # 3. 显示图片 plt.imshow(img.permute(1, 2, 0)) # 调整格式(PyTorch是[通道,高,宽],plt需要[高,宽,通道]) plt.title(f"Label: {label}") # 显示标签(CIFAR10的标签是0-9,对应不同类别) plt.show()

@浙大疏锦行

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/22 0:40:49

Zed编辑器插件生态:完整指南与开发实战

Zed编辑器插件生态:完整指南与开发实战 【免费下载链接】zed Zed 是由 Atom 和 Tree-sitter 的创造者开发的一款高性能、多人协作代码编辑器。 项目地址: https://gitcode.com/GitHub_Trending/ze/zed 想要充分发挥Zed编辑器的潜力吗?通过Zed编辑…

作者头像 李华
网站建设 2026/9/21 16:32:12

CppMicroServices 终极指南:5步掌握C++模块化开发

CppMicroServices 终极指南:5步掌握C模块化开发 【免费下载链接】CppMicroServices An OSGi-like C dynamic module system and service registry 项目地址: https://gitcode.com/gh_mirrors/cp/CppMicroServices CppMicroServices 是一个基于 OSGi 理念的 C…

作者头像 李华
网站建设 2026/9/22 0:41:40

SVG转Canvas渲染引擎终极指南:从零到精通的完整教程

SVG转Canvas渲染引擎终极指南:从零到精通的完整教程 【免费下载链接】canvg 项目地址: https://gitcode.com/gh_mirrors/can/canvg canvg是一个功能强大的JavaScript库,专门用于将SVG矢量图形解析并渲染到HTML5 Canvas画布上。通过SVG转Canvas技…

作者头像 李华
网站建设 2026/9/22 1:38:46

Langchain-Chatchat科研辅助系统构建:论文资料智能问答平台

Langchain-Chatchat科研辅助系统构建:论文资料智能问答平台 在高校实验室和科研机构中,研究人员常常面对堆积如山的PDF论文、技术报告与实验记录。如何从这些非结构化文档中快速定位关键信息?传统搜索引擎依赖关键词匹配,常因术语…

作者头像 李华
网站建设 2026/9/21 4:46:40

FastAPI蓝绿部署实战指南:实现零停机更新

FastAPI蓝绿部署实战指南:实现零停机更新 【免费下载链接】awesome-fastapi A curated list of awesome things related to FastAPI 项目地址: https://gitcode.com/gh_mirrors/aw/awesome-fastapi 在现代软件开发中,保持应用程序的高可用性至关重…

作者头像 李华