自定义数据集dataset

艾文教编程

博客专家认证

2023-01-12 21:54:25

课时名称	课时知识点
自定义数据集dataset	自定义数据集dataset

...全文

167 回复打赏收藏转发到动态举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

pytorch 自定义数据集 notebook # Dataset 类 # torch.utils.data.Dataset 是一个抽象类, 表示一个dataset. # 自定义的dataset类需要继承Dataset. 并且重载: # __len__函数, len(dataset)返回数据集的长度 # __getitem__函数, 支持dataset[i]寻址.

line,graphic,line,graphic,line,graphic

自定义数据集Dataset

自定义自己的数据集，需要继承，并且实现__len__和__getitem__方法。

在学习Tensorflow的过程中，发现大多数教程都是基于现有的数据集进行训练、优化。例如：MNIST识别教程，一个 (x_train, y_train), (x_test, y_test) = mnist.load_data() 即可获得训练、测试数据集。而在解决实际问题时，我们经常面对的是采集到的原始图片信息，这些图片保存在硬盘当中，当模型搭建好以后开始把数据从硬盘加载到内存，然...

艾文的课程社区_NO_2

3

社区成员

450

社区内容

发帖

与我相关

我的任务

社区管理员

加入社区

近7日
近30日
至今

加载中

查看更多榜单

社区公告

暂无公告

试试用AI创作助手写篇文章吧

+ 用AI写文章