当前位置：网站首页>【Pytorch】学习笔记（一）创建自己的数据集

【Pytorch】学习笔记（一）创建自己的数据集

2022-07-21 02:31:00 【高小耸】

自定义数据集
要想使用自己的数据集，必须实现3个方法: __init__(),__len__()与__getitem__().
___init__():
在初始化数据集时调用一次，我们在初始化方法里完成一些数据的初始化工作，例如，初始化图片路径，标注文件以及数据预处理的初始化。
__len__():
返回数据集中的数据数量
__getitem__():
__getitem__()的方法根据给定的idx，加载然后返回一个样本。基于idx,找到图片并转换成tensor.遍历label,调用自定义的transform函数来完成图片的预处理。最终返回tensor化的图片和对应的标签元组。

在这里插入图片描述

原网站

版权声明
本文为[高小耸]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_40271181/article/details/125842865

边栏推荐

猜你喜欢

随机推荐