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

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

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值