logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch加载数据集的方式总结

目前我们有自己制作的数据以及数据标签,但是有时候感觉不太适合直接用Pytorch自带加载数据集的方法。我们可以自己来重写定义一个类,这个类继承于,同时我们需要重写这个类里面的两个方法_getitem__()和__len()__函数。如下所示。这两种方法如何构造以及具体的细节可以查看其他的博客。len方法必须返回数据的长度,getitem方法必须返回数据以及标签。通过上述的程序,我们构造了一个数据加

#pytorch#python#深度学习
暂无文章信息