栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1019432.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号